diff --git a/invokeai/backend/image_util/lama.py b/invokeai/backend/image_util/lama.py index 9c5e0153c4..1a60fc9a05 100644 --- a/invokeai/backend/image_util/lama.py +++ b/invokeai/backend/image_util/lama.py @@ -53,5 +53,6 @@ class LaMA: del model gc.collect() + torch.cuda.empty_cache() return infilled_image