From b0686db3b66d4a0ebb0f69584b1501fc53e6e804 Mon Sep 17 00:00:00 2001 From: Arjun Suresh Date: Wed, 6 Nov 2024 22:34:14 +0000 Subject: [PATCH] Fix SDXL bug with diffuser 0.30.0 while giving downloaded model path (#1905) --- text_to_image/backend_pytorch.py | 1 - 1 file changed, 1 deletion(-) diff --git a/text_to_image/backend_pytorch.py b/text_to_image/backend_pytorch.py index 1a45c1ca5..36e2b8009 100644 --- a/text_to_image/backend_pytorch.py +++ b/text_to_image/backend_pytorch.py @@ -84,7 +84,6 @@ def load(self): scheduler=self.scheduler, safety_checker=None, add_watermarker=False, - variant="fp16" if (self.dtype == torch.float16) else None, torch_dtype=self.dtype, ) # self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)