patrickvonplaten commited on
Commit
8c3b62a
1 Parent(s): dd5eade
Files changed (1) hide show
  1. image_transformation.py +1 -3
image_transformation.py CHANGED
@@ -55,9 +55,7 @@ class ImageTransformationTool(Tool):
55
  if self.device is None:
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)
63
  if self.device.type == "cuda":
 
55
  if self.device is None:
56
  self.device = get_default_device()
57
 
58
+ self.pipeline = DiffusionPipeline.from_pretrained(self.stable_diffusion)
 
 
59
 
60
  self.pipeline.to(self.device)
61
  if self.device.type == "cuda":