Eueuiaa commited on
Commit
b1c5e84
·
verified ·
1 Parent(s): d9a4aef

Update api/ltx_server_refactored.py

Browse files
Files changed (1) hide show
  1. api/ltx_server_refactored.py +2 -2
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 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,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 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)
 
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)