Skip to content

GitLab

Unverified Commit 8c1d91e1 authored by Watebear's avatar Watebear Committed by GitHub
Browse files

[bugfix]: fix conflict (#465) 


Co-authored-by: default avatarYang Yong (雍洋) <yongyang1030@163.com>
parent ad564439
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
lightx2v/utils/set_config.py
View file @ 8c1d91e1
...@@ -71,6 +71,7 @@ def set_config(args): ...@@ -71,6 +71,7 @@ def set_config(args):
logger.warning(f"`num_frames - 1` has to be divisible by {config['vae_stride'][0]}. Rounding to the nearest number.") logger.warning(f"`num_frames - 1` has to be divisible by {config['vae_stride'][0]}. Rounding to the nearest number.")
config["target_video_length"] = config["target_video_length"] // config["vae_stride"][0] * config["vae_stride"][0] + 1 config["target_video_length"] = config["target_video_length"] // config["vae_stride"][0] * config["vae_stride"][0] + 1
if config["task"] not in ["t2i", "i2i"]:
config["attnmap_frame_num"] = ((config["target_video_length"] - 1) // config["vae_stride"][0] + 1) // config["patch_size"][0] config["attnmap_frame_num"] = ((config["target_video_length"] - 1) // config["vae_stride"][0] + 1) // config["patch_size"][0]
if config["model_cls"] == "seko_talk": if config["model_cls"] == "seko_talk":
config["attnmap_frame_num"] += 1 config["attnmap_frame_num"] += 1
......
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