Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_pipeline.py
Browse files
api/ltx/ltx_aduc_pipeline.py
CHANGED
|
@@ -158,7 +158,7 @@ class LtxAducPipeline:
|
|
| 158 |
initial_conditions = []
|
| 159 |
if initial_media_items:
|
| 160 |
logging.info("Preparing initial conditioning items from raw media list...")
|
| 161 |
-
initial_conditions =
|
| 162 |
media_items=[item[0] for item in initial_media_items],
|
| 163 |
target_frames=[item[1] for item in initial_media_items],
|
| 164 |
strengths=[item[2] for item in initial_media_items],
|
|
@@ -242,7 +242,7 @@ class LtxAducPipeline:
|
|
| 242 |
torch.save(final_latents, final_latents_path)
|
| 243 |
logging.info(f"Final latents saved to: {final_latents_path}")
|
| 244 |
|
| 245 |
-
pixel_tensor =
|
| 246 |
final_latents, decode_timestep=float(self.config.get("decode_timestep", 0.05))
|
| 247 |
)
|
| 248 |
video_path = self._save_and_log_video(pixel_tensor, f"{base_filename}_{seed}")
|
|
|
|
| 158 |
initial_conditions = []
|
| 159 |
if initial_media_items:
|
| 160 |
logging.info("Preparing initial conditioning items from raw media list...")
|
| 161 |
+
initial_conditions = vae_ltx_aduc_pipeline.generate_conditioning_items(
|
| 162 |
media_items=[item[0] for item in initial_media_items],
|
| 163 |
target_frames=[item[1] for item in initial_media_items],
|
| 164 |
strengths=[item[2] for item in initial_media_items],
|
|
|
|
| 242 |
torch.save(final_latents, final_latents_path)
|
| 243 |
logging.info(f"Final latents saved to: {final_latents_path}")
|
| 244 |
|
| 245 |
+
pixel_tensor = vae_ltx_aduc_pipeline.decode_to_pixels(
|
| 246 |
final_latents, decode_timestep=float(self.config.get("decode_timestep", 0.05))
|
| 247 |
)
|
| 248 |
video_path = self._save_and_log_video(pixel_tensor, f"{base_filename}_{seed}")
|