"torchvision/vscode:/vscode.git/clone" did not exist on "f5afae50bc8e99b873e2345bcda2dedfc863a737"
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
...@@ -71,9 +71,10 @@ def set_config(args): ...@@ -71,9 +71,10 @@ 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
config["attnmap_frame_num"] = ((config["target_video_length"] - 1) // config["vae_stride"][0] + 1) // config["patch_size"][0] if config["task"] not in ["t2i", "i2i"]:
if config["model_cls"] == "seko_talk": config["attnmap_frame_num"] = ((config["target_video_length"] - 1) // config["vae_stride"][0] + 1) // config["patch_size"][0]
config["attnmap_frame_num"] += 1 if config["model_cls"] == "seko_talk":
config["attnmap_frame_num"] += 1
return config return config
......
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