Update api/ltx_server.py
Browse files- api/ltx_server.py +2 -0
api/ltx_server.py
CHANGED
|
@@ -425,6 +425,7 @@ class VideoService:
|
|
| 425 |
|
| 426 |
|
| 427 |
if steps > 1:
|
|
|
|
| 428 |
while i < steps:
|
| 429 |
start = i*num_latente_por_chunk
|
| 430 |
end = (start+num_latente_por_chunk)-1
|
|
@@ -433,6 +434,7 @@ class VideoService:
|
|
| 433 |
chunk = latents_brutos[:, :, start:end, :, :].clone()
|
| 434 |
chunks.append(chunk)
|
| 435 |
print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
|
|
|
|
| 436 |
else:
|
| 437 |
print(f"[DEBUG] numero chunks minimo")
|
| 438 |
print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")
|
|
|
|
| 425 |
|
| 426 |
|
| 427 |
if steps > 1:
|
| 428 |
+
i = 0
|
| 429 |
while i < steps:
|
| 430 |
start = i*num_latente_por_chunk
|
| 431 |
end = (start+num_latente_por_chunk)-1
|
|
|
|
| 434 |
chunk = latents_brutos[:, :, start:end, :, :].clone()
|
| 435 |
chunks.append(chunk)
|
| 436 |
print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
|
| 437 |
+
i += 1
|
| 438 |
else:
|
| 439 |
print(f"[DEBUG] numero chunks minimo")
|
| 440 |
print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")
|