patrickvonplaten commited on
Commit
dd5eade
1 Parent(s): c3c0857
Files changed (1) hide show
  1. image_transformation.py +1 -1
image_transformation.py CHANGED
@@ -56,7 +56,7 @@ class ImageTransformationTool(Tool):
56
  self.device = get_default_device()
57
 
58
  self.pipeline = DiffusionPipeline.from_pretrained(
59
- self.stable_diffusion_checkpoint, controlnet=self.controlnet
60
  )
61
 
62
  self.pipeline.to(self.device)
 
56
  self.device = get_default_device()
57
 
58
  self.pipeline = DiffusionPipeline.from_pretrained(
59
+ self.stable_diffusion, controlnet=self.controlnet
60
  )
61
 
62
  self.pipeline.to(self.device)