update

2023-12-19 13:16:30 +08:00
parent f27fc51e34
commit 141936a937
18 changed files with 479 additions and 358 deletions
--- a/lama_cleaner/model/sdxl.py
+++ b/lama_cleaner/model/sdxl.py
@@ -39,7 +39,7 @@ class SDXL(DiffusionInpaintModel):
            )
        else:
            vae = AutoencoderKL.from_pretrained(
-                "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
+                "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch_dtype
            )
            self.model = StableDiffusionXLInpaintPipeline.from_pretrained(
                self.model_id_or_path,