EuuIia commited on
Commit
0f11d75
·
verified ·
1 Parent(s): cf48f76

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. 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.15,
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, 2)
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)