EuuIia commited on
Commit
f0b5401
·
verified ·
1 Parent(s): b6e3b11

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. 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[^18_0])
67
- name = parts[^18_1]
68
- used_mb = int(parts[^18_2])
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[^18_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)
 
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)