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)