Fix SDXL bug with diffuser 0.30.0 while giving downloaded model path (mlcommons#1905)

arjunsuresh · web-flow · commit b0686db3b66d · 2024-11-06T22:34:14.000Z
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)

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,6 @@ def load(self):`
`84`	`84`	`scheduler=self.scheduler,`
`85`	`85`	`safety_checker=None,`
`86`	`86`	`add_watermarker=False,`
`87`		`- variant="fp16" if (self.dtype == torch.float16) else None,`
`88`	`87`	`torch_dtype=self.dtype,`
`89`	`88`	`)`
`90`	`89`	`# self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)`