auto switch mps device to cpu device

2023-02-11 13:30:09 +08:00
parent f9b5dcbfd7
commit 8f8bcfe0f4
15 changed files with 52 additions and 19 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -160,8 +160,10 @@ class SD(DiffusionInpaintModel):


 class SD15(SD):
+    name = "sd1.5"
    model_id_or_path = "runwayml/stable-diffusion-inpainting"


 class SD2(SD):
+    name = "sd2"
    model_id_or_path = "stabilityai/stable-diffusion-2-inpainting"