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,
|