Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -104,7 +104,7 @@ def FlowEditRun(
|
|
| 104 |
raise NotImplementedError(f"Model type {model_type} not implemented")
|
| 105 |
|
| 106 |
scheduler = pipe.scheduler
|
| 107 |
-
pipe = pipe.to(
|
| 108 |
|
| 109 |
|
| 110 |
|
|
|
|
| 104 |
raise NotImplementedError(f"Model type {model_type} not implemented")
|
| 105 |
|
| 106 |
scheduler = pipe.scheduler
|
| 107 |
+
pipe = pipe.to("cuda")
|
| 108 |
|
| 109 |
|
| 110 |
|