Update api/ltx_server.py
Browse files- api/ltx_server.py +3 -3
api/ltx_server.py
CHANGED
|
@@ -420,16 +420,16 @@ class VideoService:
|
|
| 420 |
print(f"[DEBUG] TOTAL LATENTES A DIVIDIR /{quantidade} = {sum_latent}")
|
| 421 |
|
| 422 |
for i in range(quantidade):
|
| 423 |
-
start = i * step
|
| 424 |
end = start + step
|
| 425 |
if i == quantidade - 1:
|
| 426 |
-
end =
|
| 427 |
else:
|
| 428 |
end += overlap
|
| 429 |
chunk = latents_brutos[:, :, start:end, :, :].clone()
|
| 430 |
chunks.append(chunk)
|
| 431 |
chunk_sum = chunk.shape[2]
|
| 432 |
-
print(f"[DEBUG] chunk{i+1}[:, :, start:end, :, :] = {chunk_sum}")
|
| 433 |
|
| 434 |
print(f"================PODA CAUSAL=================")
|
| 435 |
return chunks
|
|
|
|
| 420 |
print(f"[DEBUG] TOTAL LATENTES A DIVIDIR /{quantidade} = {sum_latent}")
|
| 421 |
|
| 422 |
for i in range(quantidade):
|
| 423 |
+
start = i * step
|
| 424 |
end = start + step
|
| 425 |
if i == quantidade - 1:
|
| 426 |
+
end = sum_latent
|
| 427 |
else:
|
| 428 |
end += overlap
|
| 429 |
chunk = latents_brutos[:, :, start:end, :, :].clone()
|
| 430 |
chunks.append(chunk)
|
| 431 |
chunk_sum = chunk.shape[2]
|
| 432 |
+
print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk_sum}")
|
| 433 |
|
| 434 |
print(f"================PODA CAUSAL=================")
|
| 435 |
return chunks
|