PengWeixuanSZU commited on
Commit
ec765c7
·
verified ·
1 Parent(s): 886e420

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=pipe.to("cuda")
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,