Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -484,7 +484,7 @@ if __name__ == "__main__":
|
|
484 |
# scheduler=eulera_scheduler,
|
485 |
)
|
486 |
pipe_control.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_control.scheduler.config)
|
487 |
-
pipe_control.cuda
|
488 |
|
489 |
# try:
|
490 |
# pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|
|
|
484 |
# scheduler=eulera_scheduler,
|
485 |
)
|
486 |
pipe_control.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_control.scheduler.config)
|
487 |
+
pipe_control.to("cuda")
|
488 |
|
489 |
# try:
|
490 |
# pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|