Spaces:
Build error
Build error
patrickvonplaten
commited on
Commit
•
dd5eade
1
Parent(s):
c3c0857
fix more
Browse files- 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.
|
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)
|