EuuIia commited on
Commit
9ea7873
·
verified ·
1 Parent(s): 9b10e93

Upload ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +4 -4
api/ltx_server.py CHANGED
@@ -65,7 +65,7 @@ def _query_gpu_processes_via_nvidiasmi(device_index: int) -> List[Dict]:
65
  parts = [p.strip() for p in line.split(",")]
66
  if len(parts) >= 3:
67
  try:
68
- pid = int(parts[^23_0]); name = parts[^23_1]; used_mb = int(parts[^23_2])
69
  user = "unknown"
70
  try:
71
  import psutil
@@ -373,7 +373,7 @@ class VideoService:
373
  img_bchw = self.pipeline.vae.decode(latent_chw.unsqueeze(0))
374
  else:
375
  raise RuntimeError("Nenhum decoder (decode_latents/vae.decode) disponível.")
376
- img_chw = img_bchw[^23_0]
377
  if img_chw.min() < 0:
378
  img_chw = (img_chw.clamp(-1, 1) + 1.0) / 2.0
379
  else:
@@ -385,7 +385,7 @@ class VideoService:
385
  Converte (C,H,W) float [0,1] em (H,W,C) uint8 com crop do padding.
386
  """
387
  pad_left, pad_right, pad_top, pad_bottom = padding_values
388
- H, W = pixel_chw.shape[^23_1], pixel_chw.shape[^23_2]
389
  h_end = H - pad_bottom if pad_bottom > 0 else H
390
  w_end = W - pad_right if pad_right > 0 else W
391
  pixel_chw = pixel_chw[:, pad_top:h_end, pad_left:w_end]
@@ -401,7 +401,7 @@ class VideoService:
401
  """
402
  Latentes (B,C,T,H,W) → decodifica quadro a quadro → escreve MP4 incremental.
403
  """
404
- T = latents.shape[^23_2]
405
  print(f"[DEBUG] encode_latents_to_mp4: frames={T} out={output_path}")
406
  with imageio.get_writer(output_path, fps=fps, codec="libx264", quality=8) as writer:
407
  for i in range(T):
 
65
  parts = [p.strip() for p in line.split(",")]
66
  if len(parts) >= 3:
67
  try:
68
+ pid = int(parts[0]); name = parts[_1]; used_mb = int(parts[_2])
69
  user = "unknown"
70
  try:
71
  import psutil
 
373
  img_bchw = self.pipeline.vae.decode(latent_chw.unsqueeze(0))
374
  else:
375
  raise RuntimeError("Nenhum decoder (decode_latents/vae.decode) disponível.")
376
+ img_chw = img_bchw[0]
377
  if img_chw.min() < 0:
378
  img_chw = (img_chw.clamp(-1, 1) + 1.0) / 2.0
379
  else:
 
385
  Converte (C,H,W) float [0,1] em (H,W,C) uint8 com crop do padding.
386
  """
387
  pad_left, pad_right, pad_top, pad_bottom = padding_values
388
+ H, W = pixel_chw.shape[1], pixel_chw.shape[2]
389
  h_end = H - pad_bottom if pad_bottom > 0 else H
390
  w_end = W - pad_right if pad_right > 0 else W
391
  pixel_chw = pixel_chw[:, pad_top:h_end, pad_left:w_end]
 
401
  """
402
  Latentes (B,C,T,H,W) → decodifica quadro a quadro → escreve MP4 incremental.
403
  """
404
+ T = latents.shape[2]
405
  print(f"[DEBUG] encode_latents_to_mp4: frames={T} out={output_path}")
406
  with imageio.get_writer(output_path, fps=fps, codec="libx264", quality=8) as writer:
407
  for i in range(T):