alexnasa commited on
Commit
e592ff0
·
verified ·
1 Parent(s): 8764c13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -14
app.py CHANGED
@@ -400,22 +400,12 @@ class WanInferencePipeline(nn.Module):
400
  h = height
401
  w = width
402
 
403
- # step 1: numerator and denominator as ints
404
  num = args.max_tokens * 16 * 16 * 4
405
  den = h * w
406
-
407
- # step 2: integer division
408
- L0 = num // den # exact floor division, no float in sight
409
-
410
- # step 3: make it ≡ 1 mod 4
411
- diff = (L0 - 1) % 4
412
- L = L0 - diff
413
- if L < 1:
414
- L = 1 # or whatever your minimal frame count is
415
-
416
- # step 4: latent frames
417
- T = (L + 3) // 4
418
-
419
 
420
  if self.args.i2v:
421
  if self.args.random_prefix_frames:
 
400
  h = height
401
  w = width
402
 
 
403
  num = args.max_tokens * 16 * 16 * 4
404
  den = h * w
405
+ L0 = num // den
406
+ L = L0 - (L0 % 4)
407
+ if L < 4: L = 4
408
+ T = L // 4
 
 
 
 
 
 
 
 
 
409
 
410
  if self.args.i2v:
411
  if self.args.random_prefix_frames: