Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
| 406 |
-
|
| 407 |
-
|
| 408 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|