Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -198,7 +198,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
|
|
| 198 |
height = 832
|
| 199 |
width = 480
|
| 200 |
|
| 201 |
-
pipe
|
| 202 |
with torch.no_grad():
|
| 203 |
out = pipe(
|
| 204 |
images=img_tensor,
|
|
|
|
| 198 |
height = 832
|
| 199 |
width = 480
|
| 200 |
|
| 201 |
+
pipe.to("cuda")
|
| 202 |
with torch.no_grad():
|
| 203 |
out = pipe(
|
| 204 |
images=img_tensor,
|