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/controlnet.py
+++ b/iopaint/model/controlnet.py
@@ -111,7 +111,7 @@ class ControlNet(DiffusionInpaintModel):
                pretrained_model_name_or_path=model_info.path,
                controlnet=controlnet,
                variant="fp16",
-                dtype=torch_dtype,
+                torch_dtype=torch_dtype,
                **model_kwargs,
            )