Spaces:

NKU-AMT
/

AMT

Runtime error

zzl commited on Apr 19, 2023

Commit

1947ad8

1 Parent(s): 981d2df

release

Files changed (1) hide show

demo_vid.py CHANGED Viewed

@@ -46,7 +46,7 @@ def vid2vid(model_type, video, iters):
     if scale < 1:
         print(f"Due to the limited VRAM, the video will be scaled by {scale:.2f}")
     padding = int(16 / scale)
-    padder = InputPadder(inputs[0].shape, padding)
     while True:
         ret, frame = vcap.read()
         if ret is False:

     if scale < 1:
         print(f"Due to the limited VRAM, the video will be scaled by {scale:.2f}")
     padding = int(16 / scale)
+    padder = InputPadder((h, w), padding)
     while True:
         ret, frame = vcap.read()
         if ret is False: