Update api/ltx_server.py
Browse files- api/ltx_server.py +4 -4
api/ltx_server.py
CHANGED
|
@@ -63,9 +63,9 @@ def _query_gpu_processes_via_nvidiasmi(device_index: int) -> List[Dict]:
|
|
| 63 |
parts = [p.strip() for p in line.split(",")]
|
| 64 |
if len(parts) >= 3:
|
| 65 |
try:
|
| 66 |
-
pid = int(parts[
|
| 67 |
-
name = parts[
|
| 68 |
-
used_mb = int(parts[
|
| 69 |
user = "unknown"
|
| 70 |
try:
|
| 71 |
import psutil
|
|
@@ -515,7 +515,7 @@ class VideoService:
|
|
| 515 |
try:
|
| 516 |
# Escrita quadro a quadro para evitar array 4D gigante em RAM
|
| 517 |
with imageio.get_writer(output_video_path, fps=call_kwargs["frame_rate"], codec="libx264", quality=8) as writer:
|
| 518 |
-
T = result_tensor.shape[
|
| 519 |
for i in range(T):
|
| 520 |
frame_chw = result_tensor[0, :, i] # (C,H,W) no device
|
| 521 |
frame_hwc_u8 = (frame_chw.permute(1, 2, 0) # (H,W,C)
|
|
|
|
| 63 |
parts = [p.strip() for p in line.split(",")]
|
| 64 |
if len(parts) >= 3:
|
| 65 |
try:
|
| 66 |
+
pid = int(parts[0])
|
| 67 |
+
name = parts[1]
|
| 68 |
+
used_mb = int(parts[2])
|
| 69 |
user = "unknown"
|
| 70 |
try:
|
| 71 |
import psutil
|
|
|
|
| 515 |
try:
|
| 516 |
# Escrita quadro a quadro para evitar array 4D gigante em RAM
|
| 517 |
with imageio.get_writer(output_video_path, fps=call_kwargs["frame_rate"], codec="libx264", quality=8) as writer:
|
| 518 |
+
T = result_tensor.shape[2] # (B, C, T, H, W)
|
| 519 |
for i in range(T):
|
| 520 |
frame_chw = result_tensor[0, :, i] # (C,H,W) no device
|
| 521 |
frame_hwc_u8 = (frame_chw.permute(1, 2, 0) # (H,W,C)
|