Commit a9f0ab52 authored by helloyongyang's avatar helloyongyang
Browse files

update log

parent a540357c
......@@ -233,13 +233,13 @@ class DefaultRunner(BaseRunner):
fps = self.config.get("fps", 16)
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":
save_to_video(images, self.config.save_video_path, fps=fps, method="ffmpeg") # type: ignore
else:
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
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