Spaces:
Paused
Paused
Update api/ltx_server_refactored.py
Browse files
api/ltx_server_refactored.py
CHANGED
|
@@ -295,7 +295,7 @@ class VideoService:
|
|
| 295 |
"output_type": "latent", "conditioning_items": conditions_itens,
|
| 296 |
**(self.config.get("first_pass", {}))
|
| 297 |
}
|
| 298 |
-
if
|
| 299 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 300 |
with ctx:
|
| 301 |
latents = self.pipeline(**first_pass_kwargs).images
|
|
@@ -393,7 +393,7 @@ class VideoService:
|
|
| 393 |
#tensor_path = os.path.join(results_dir, f"latents_narrative_{used_seed}.pt")
|
| 394 |
#torch.save(final_latents_cpu, tensor_path)
|
| 395 |
|
| 396 |
-
if
|
| 397 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 398 |
with ctx:
|
| 399 |
final_latents_video = torch.cat(latentes_chunk_video_list, dim=2)
|
|
|
|
| 295 |
"output_type": "latent", "conditioning_items": conditions_itens,
|
| 296 |
**(self.config.get("first_pass", {}))
|
| 297 |
}
|
| 298 |
+
if True:
|
| 299 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 300 |
with ctx:
|
| 301 |
latents = self.pipeline(**first_pass_kwargs).images
|
|
|
|
| 393 |
#tensor_path = os.path.join(results_dir, f"latents_narrative_{used_seed}.pt")
|
| 394 |
#torch.save(final_latents_cpu, tensor_path)
|
| 395 |
|
| 396 |
+
if True:
|
| 397 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 398 |
with ctx:
|
| 399 |
final_latents_video = torch.cat(latentes_chunk_video_list, dim=2)
|