removed cpu option for lama cleaner

jahangir091 · Dec 10, 2023 · e824361 · e824361
1 parent 5530325
commit e824361
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/extensions/sd-webui-cleaner/scripts/lama.py b/extensions/sd-webui-cleaner/scripts/lama.py
@@ -36,9 +36,6 @@ def clean_object(image,mask):
         result = Lama.predict(init_image, mask_image)
     except:
         pass
-    finally:
-        print('############# lama using CPU ###########################')
-        Lama.to("cpu")
 
     return [result]