fix https://github.com/Sanster/IOPaint/issues/508

diffusers removed config_files api: https://github.com/huggingface/diffusers/issues/6819#issuecomment-1928396112
2024-04-12 13:09:37 +08:00
parent 21f17c8e2a
commit bfad842a0e
4 changed files with 13 additions and 9 deletions
--- a/iopaint/model/sd.py
+++ b/iopaint/model/sd.py
@@ -52,7 +52,7 @@ class SD(DiffusionInpaintModel):
                self.model_id_or_path,
                torch_dtype=torch_dtype,
                load_safety_checker=not disable_nsfw_checker,
-                config_files=get_config_files(),
+                original_config_file=get_config_files()['v1'],
                **model_kwargs,
            )
        else: