better handle scan single file diffusion

2024-04-12 18:52:33 +08:00
parent 0a262fa811
commit 76823355fe
3 changed files with 7 additions and 10 deletions
--- a/iopaint/model/brushnet/brushnet_wrapper.py
+++ b/iopaint/model/brushnet/brushnet_wrapper.py
@@ -64,7 +64,7 @@ class BrushNetWrapper(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'],
                brushnet=brushnet,
                **model_kwargs,
            )