fix torch_dtype & cpu_text_encoder

2024-02-10 10:59:06 +08:00
parent 180b7d6c70
commit 9aa5a7e0ba
4 changed files with 13 additions and 4 deletions
--- a/iopaint/model/sdxl.py
+++ b/iopaint/model/sdxl.py
@@ -39,7 +39,7 @@ class SDXL(DiffusionInpaintModel):
        if os.path.isfile(self.model_id_or_path):
            self.model = StableDiffusionXLInpaintPipeline.from_single_file(
                self.model_id_or_path,
-                dtype=torch_dtype,
+                torch_dtype=torch_dtype,
                num_in_channels=num_in_channels,
                load_safety_checker=False,
                config_files=get_config_files()