from_single_file load_safety_checker work

2024-01-10 21:22:59 +08:00
parent 772ef65f7b
commit cbcdf3b9a2
3 changed files with 16 additions and 5 deletions
--- a/iopaint/model/sdxl.py
+++ b/iopaint/model/sdxl.py
@@ -35,6 +35,7 @@ class SDXL(DiffusionInpaintModel):
                self.model_id_or_path,
                dtype=torch_dtype,
                num_in_channels=num_in_channels,
+                load_safety_checker=False
            )
        else:
            model_kwargs = {**kwargs.get("pipe_components", {})}