Spaces:
Build error
Build error
RamAnanth1
commited on
Commit
•
a9d7f19
1
Parent(s):
ee727f1
Update app.py
Browse files
app.py
CHANGED
@@ -237,7 +237,7 @@ class canny2image:
|
|
237 |
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
238 |
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
|
239 |
)
|
240 |
-
self.pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
241 |
|
242 |
# This command loads the individual model components on GPU on-demand. So, we don't
|
243 |
# need to explicitly call pipe.to("cuda").
|
|
|
237 |
self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
238 |
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
|
239 |
)
|
240 |
+
self.pipe.scheduler = UniPCMultistepScheduler.from_config(self.pipe.scheduler.config)
|
241 |
|
242 |
# This command loads the individual model components on GPU on-demand. So, we don't
|
243 |
# need to explicitly call pipe.to("cuda").
|