euiiiia commited on
Commit
5e3ff4a
·
verified ·
1 Parent(s): 7fa9709

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
@@ -401,8 +401,8 @@ class VideoService:
401
  }
402
 
403
  #with torch.autocast(device_type=self.device.split(':')[0], dtype=self.runtime_autocast_dtype, enabled=(self.device == 'cuda')):
404
- latents = self.pipeline(**second_pass_kwargs).images
405
- pixel_tensor = vae_manager_singleton.decode(latents.clone(), decode_timestep=float(self.config.get("decode_timestep", 0.05)))
406
  video_path = self._save_video_from_tensor(pixel_tensor, "low_res_video", used_seed, temp_dir)
407
  latents_path = self._save_latents_to_disk(latents, "latents_low_res", used_seed)
408
 
 
401
  }
402
 
403
  #with torch.autocast(device_type=self.device.split(':')[0], dtype=self.runtime_autocast_dtype, enabled=(self.device == 'cuda')):
404
+ #latents = self.pipeline(**second_pass_kwargs).images
405
+ pixel_tensor = vae_manager_singleton.decode(latents_low.clone(), decode_timestep=float(self.config.get("decode_timestep", 0.05)))
406
  video_path = self._save_video_from_tensor(pixel_tensor, "low_res_video", used_seed, temp_dir)
407
  latents_path = self._save_latents_to_disk(latents, "latents_low_res", used_seed)
408