alexnasa commited on
Commit
6b2fdf5
·
verified ·
1 Parent(s): e592ff0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -400,12 +400,22 @@ class WanInferencePipeline(nn.Module):
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:
 
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: