Spaces:

huggingface-tools
/

image-transformation

Running

patrickvonplaten commited on May 5, 2023

Commit

f5d0858

•

1 Parent(s): ed1d8d0

fix

Files changed (1) hide show

image_transformation.py CHANGED Viewed

@@ -68,7 +68,7 @@ class ImageTransformationTool(Tool):
         )
         self.pipeline.scheduler = UniPCMultistepScheduler.from_config(self.pipeline.scheduler.config)
-        self.pipeline.to(device=device)
         if self.device.type == "cuda":
             self.pipeline.to(torch_dtype=torch.float16)

         )
         self.pipeline.scheduler = UniPCMultistepScheduler.from_config(self.pipeline.scheduler.config)
+        self.pipeline.to(device=self.device)
         if self.device.type == "cuda":
             self.pipeline.to(torch_dtype=torch.float16)