Skip to content

GitLab

Commit a9f0ab52 authored by helloyongyang's avatar helloyongyang
Browse files

update log

parent a540357c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lightx2v/models/runners/default_runner.py
View file @ a9f0ab52
...@@ -233,13 +233,13 @@ class DefaultRunner(BaseRunner): ...@@ -233,13 +233,13 @@ class DefaultRunner(BaseRunner):
fps = self.config.get("fps", 16) fps = self.config.get("fps", 16)
if not dist.is_initialized() or dist.get_rank() == 0: if not dist.is_initialized() or dist.get_rank() == 0:
logger.info(f"Saving video to {self.config.save_video_path}") logger.info(f"🎬 Start to save video 🎬")
if self.config["model_cls"] != "wan2.2": if self.config["model_cls"] != "wan2.2":
save_to_video(images, self.config.save_video_path, fps=fps, method="ffmpeg") # type: ignore save_to_video(images, self.config.save_video_path, fps=fps, method="ffmpeg") # type: ignore
else: else:
cache_video(tensor=images, save_file=self.config.save_video_path, fps=fps, nrow=1, normalize=True, value_range=(-1, 1)) cache_video(tensor=images, save_file=self.config.save_video_path, fps=fps, nrow=1, normalize=True, value_range=(-1, 1))
logger.info(f"Video saved successfully.") logger.info(f"✅ Video saved successfully to: {self.config.save_video_path} ✅")
del latents, generator del latents, generator
torch.cuda.empty_cache() torch.cuda.empty_cache()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment