Upload ltx_server.py
Browse files- 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[
|
| 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[
|
| 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[
|
| 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[
|
| 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):
|