Update api/ltx_server.py
Browse files- api/ltx_server.py +5 -6
api/ltx_server.py
CHANGED
|
@@ -278,9 +278,8 @@ class VideoService:
|
|
| 278 |
def _load_config(self):
|
| 279 |
base = LTX_VIDEO_REPO_DIR / "configs"
|
| 280 |
candidates = [
|
| 281 |
-
base / "ltxv-13b-0.9.8-dev-fp8.yaml",
|
| 282 |
base / "ltxv-13b-0.9.8-distilled-fp8.yaml",
|
| 283 |
-
base / "ltxv-13b-0.9.8-dev-fp8.yaml.txt",
|
| 284 |
base / "ltxv-13b-0.9.8-distilled.yaml",
|
| 285 |
]
|
| 286 |
for cfg in candidates:
|
|
@@ -288,7 +287,7 @@ class VideoService:
|
|
| 288 |
print(f"[DEBUG] Config selecionada: {cfg}")
|
| 289 |
with open(cfg, "r") as file:
|
| 290 |
return yaml.safe_load(file)
|
| 291 |
-
cfg = base / "ltxv-13b-0.9.8-distilled.yaml"
|
| 292 |
print(f"[DEBUG] Config fallback: {cfg}")
|
| 293 |
with open(cfg, "r") as file:
|
| 294 |
return yaml.safe_load(file)
|
|
@@ -543,13 +542,13 @@ class VideoService:
|
|
| 543 |
"decode_timestep": self.config["decode_timestep"],
|
| 544 |
"decode_noise_scale": self.config["decode_noise_scale"],
|
| 545 |
"stochastic_sampling": self.config["stochastic_sampling"],
|
| 546 |
-
"image_cond_noise_scale": 0.
|
| 547 |
"is_video": True,
|
| 548 |
"vae_per_channel_normalize": True,
|
| 549 |
"mixed_precision": (self.config["precision"] == "mixed_precision"),
|
| 550 |
"offload_to_cpu": False,
|
| 551 |
"enhance_prompt": False,
|
| 552 |
-
"skip_layer_strategy": SkipLayerStrategy.AttentionValues,
|
| 553 |
}
|
| 554 |
print(f"[DEBUG] output_type={call_kwargs['output_type']} skip_layer_strategy={call_kwargs['skip_layer_strategy']}")
|
| 555 |
|
|
@@ -645,7 +644,7 @@ class VideoService:
|
|
| 645 |
|
| 646 |
# 2) Divide em duas partes
|
| 647 |
#lat_a, lat_b = self._dividir_latentes_em_partes(latents_cpu, 2)
|
| 648 |
-
latents_parts = self._dividir_latentes_em_partes(latents_cpu,
|
| 649 |
|
| 650 |
|
| 651 |
temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
|
|
|
|
| 278 |
def _load_config(self):
|
| 279 |
base = LTX_VIDEO_REPO_DIR / "configs"
|
| 280 |
candidates = [
|
| 281 |
+
#base / "ltxv-13b-0.9.8-dev-fp8.yaml",
|
| 282 |
base / "ltxv-13b-0.9.8-distilled-fp8.yaml",
|
|
|
|
| 283 |
base / "ltxv-13b-0.9.8-distilled.yaml",
|
| 284 |
]
|
| 285 |
for cfg in candidates:
|
|
|
|
| 287 |
print(f"[DEBUG] Config selecionada: {cfg}")
|
| 288 |
with open(cfg, "r") as file:
|
| 289 |
return yaml.safe_load(file)
|
| 290 |
+
cfg = base / "ltxv-13b-0.9.8-distilled-fp8.yaml"
|
| 291 |
print(f"[DEBUG] Config fallback: {cfg}")
|
| 292 |
with open(cfg, "r") as file:
|
| 293 |
return yaml.safe_load(file)
|
|
|
|
| 542 |
"decode_timestep": self.config["decode_timestep"],
|
| 543 |
"decode_noise_scale": self.config["decode_noise_scale"],
|
| 544 |
"stochastic_sampling": self.config["stochastic_sampling"],
|
| 545 |
+
"image_cond_noise_scale": 0.01,
|
| 546 |
"is_video": True,
|
| 547 |
"vae_per_channel_normalize": True,
|
| 548 |
"mixed_precision": (self.config["precision"] == "mixed_precision"),
|
| 549 |
"offload_to_cpu": False,
|
| 550 |
"enhance_prompt": False,
|
| 551 |
+
#"skip_layer_strategy": SkipLayerStrategy.AttentionValues,
|
| 552 |
}
|
| 553 |
print(f"[DEBUG] output_type={call_kwargs['output_type']} skip_layer_strategy={call_kwargs['skip_layer_strategy']}")
|
| 554 |
|
|
|
|
| 644 |
|
| 645 |
# 2) Divide em duas partes
|
| 646 |
#lat_a, lat_b = self._dividir_latentes_em_partes(latents_cpu, 2)
|
| 647 |
+
latents_parts = self._dividir_latentes_em_partes(latents_cpu, 4)
|
| 648 |
|
| 649 |
|
| 650 |
temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
|