Commit
·
8c3b62a
1
Parent(s):
dd5eade
fix more
Browse files- 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":
|