Fix SDXL bug with diffuser 0.30.0 while giving downloaded model path (#…

…1905)
mlcommons · Nov 6, 2024 · b0686db · b0686db
1 parent 6dcaae8
commit b0686db
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)