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/instruct_pix2pix.py
+++ b/lama_cleaner/model/instruct_pix2pix.py
@@ -9,6 +9,7 @@ from lama_cleaner.schema import Config


 class InstructPix2Pix(DiffusionInpaintModel):
+    name = "instruct_pix2pix"
    pad_mod = 8
    min_size = 512