Spaces:

huggingface-tools
/

image-transformation

Running

patrickvonplaten commited on May 5, 2023

Commit

dd5eade

1 Parent(s): c3c0857

fix more

Files changed (1) hide show

image_transformation.py CHANGED Viewed

@@ -56,7 +56,7 @@ class ImageTransformationTool(Tool):
             self.device = get_default_device()
         self.pipeline = DiffusionPipeline.from_pretrained(
-            self.stable_diffusion_checkpoint, controlnet=self.controlnet
         )
         self.pipeline.to(self.device)

             self.device = get_default_device()
         self.pipeline = DiffusionPipeline.from_pretrained(
+            self.stable_diffusion, controlnet=self.controlnet
         )
         self.pipeline.to(self.device)