Spaces:
Paused
Paused
Update api/ltx_server.py
Browse files- api/ltx_server.py +8 -4
api/ltx_server.py
CHANGED
|
@@ -578,14 +578,18 @@ class VideoService:
|
|
| 578 |
if n_chunks > 1:
|
| 579 |
i=0
|
| 580 |
while i < n_chunks:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 581 |
start = (num_latente_por_chunk*i)
|
| 582 |
end = (start+num_latente_por_chunk+(overlap+1))
|
| 583 |
if i+1 < n_chunks:
|
| 584 |
-
chunk = latents_brutos[:, :, start:end, :, :].clone().detach()
|
| 585 |
-
print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
|
| 586 |
else:
|
| 587 |
-
chunk = latents_brutos[:, :, start:, :, :].clone().detach()
|
| 588 |
-
print(f"[DEBUG] chunk{i+1}[:, :, {start}:, :, :] = {chunk.shape[2]}")
|
| 589 |
chunks.append(chunk)
|
| 590 |
i+=1
|
| 591 |
else:
|
|
|
|
| 578 |
if n_chunks > 1:
|
| 579 |
i=0
|
| 580 |
while i < n_chunks:
|
| 581 |
+
if i>0:
|
| 582 |
+
dow=1
|
| 583 |
+
else:
|
| 584 |
+
dow=0
|
| 585 |
start = (num_latente_por_chunk*i)
|
| 586 |
end = (start+num_latente_por_chunk+(overlap+1))
|
| 587 |
if i+1 < n_chunks:
|
| 588 |
+
chunk = latents_brutos[:, :, start-(dow):end, :, :].clone().detach()
|
| 589 |
+
print(f"[DEBUG] chunk{i+1}[:, :, {start-dow}:{end}, :, :] = {chunk.shape[2]}")
|
| 590 |
else:
|
| 591 |
+
chunk = latents_brutos[:, :, start-(dow):, :, :].clone().detach()
|
| 592 |
+
print(f"[DEBUG] chunk{i+1}[:, :, {start-(dow)}:, :, :] = {chunk.shape[2]}")
|
| 593 |
chunks.append(chunk)
|
| 594 |
i+=1
|
| 595 |
else:
|