Update api/ltx_server.py
Browse files- api/ltx_server.py +3 -3
api/ltx_server.py
CHANGED
|
@@ -47,7 +47,7 @@ import torch.nn.functional as F
|
|
| 47 |
|
| 48 |
# Singletons (versões simples)
|
| 49 |
from managers.vae_manager import vae_manager_singleton
|
| 50 |
-
from tools.
|
| 51 |
|
| 52 |
# --- 2. GERENCIAMENTO DE DEPENDÊNCIAS E SETUP ---
|
| 53 |
def _query_gpu_processes_via_nvml(device_index: int) -> List[Dict]:
|
|
@@ -790,13 +790,13 @@ class VideoService:
|
|
| 790 |
"decode_timestep": self.config["decode_timestep"],
|
| 791 |
"decode_noise_scale": self.config["decode_noise_scale"],
|
| 792 |
"stochastic_sampling": self.config["stochastic_sampling"],
|
| 793 |
-
"image_cond_noise_scale": 0.
|
| 794 |
"is_video": True,
|
| 795 |
"vae_per_channel_normalize": True,
|
| 796 |
"mixed_precision": (self.config["precision"] == "mixed_precision"),
|
| 797 |
"offload_to_cpu": False,
|
| 798 |
"enhance_prompt": False,
|
| 799 |
-
"skip_layer_strategy": SkipLayerStrategy.AttentionValues,
|
| 800 |
}
|
| 801 |
print(f"[DEBUG] output_type={call_kwargs['output_type']} skip_layer_strategy={call_kwargs['skip_layer_strategy']}")
|
| 802 |
|
|
|
|
| 47 |
|
| 48 |
# Singletons (versões simples)
|
| 49 |
from managers.vae_manager import vae_manager_singleton
|
| 50 |
+
from tools.video_encode_tool import video_encode_tool_singleton
|
| 51 |
|
| 52 |
# --- 2. GERENCIAMENTO DE DEPENDÊNCIAS E SETUP ---
|
| 53 |
def _query_gpu_processes_via_nvml(device_index: int) -> List[Dict]:
|
|
|
|
| 790 |
"decode_timestep": self.config["decode_timestep"],
|
| 791 |
"decode_noise_scale": self.config["decode_noise_scale"],
|
| 792 |
"stochastic_sampling": self.config["stochastic_sampling"],
|
| 793 |
+
"image_cond_noise_scale": 0.001,
|
| 794 |
"is_video": True,
|
| 795 |
"vae_per_channel_normalize": True,
|
| 796 |
"mixed_precision": (self.config["precision"] == "mixed_precision"),
|
| 797 |
"offload_to_cpu": False,
|
| 798 |
"enhance_prompt": False,
|
| 799 |
+
"skip_layer_strategy": None #SkipLayerStrategy.AttentionValues,
|
| 800 |
}
|
| 801 |
print(f"[DEBUG] output_type={call_kwargs['output_type']} skip_layer_strategy={call_kwargs['skip_layer_strategy']}")
|
| 802 |
|