alexnasa commited on
Commit
f85b5fd
·
verified ·
1 Parent(s): 69d77bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -403,8 +403,7 @@ class WanInferencePipeline(nn.Module):
403
 
404
  else:
405
  image = None
406
- h = height
407
- w = width
408
 
409
  # step 1: numerator and denominator as ints
410
  num = args.max_tokens * 16 * 16 * 4
 
403
 
404
  else:
405
  image = None
406
+ select_size = [height, width]
 
407
 
408
  # step 1: numerator and denominator as ints
409
  num = args.max_tokens * 16 * 16 * 4