Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-20 06:02:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: Zengtudor:1aab25d613c89812dbcb96d7296f68aa79b45c16
Branches Tags
Zengtudor:main
Zengtudor:dev/format
...
compare: Zengtudor:8e6dbce161b744ebf01404871c8c3d7b83d67fd9
Branches Tags
Zengtudor:main
Zengtudor:dev/format

3 Commits
1aab25d613 ... 8e6dbce161

Author SHA1 Message Date
Zhen Han
8e6dbce161
Merge e649b328e9 into 841fe5237b 2025-12-07 23:29:13 +08:00
Shiguang Ai
841fe5237b
When frame_num is changed from 81 the tensor does not match (#533) 2025-12-02 14:53:33 +08:00
hanzhn
e649b328e9 shrink vace seed range 2025-05-15 13:13:11 +08:00
2 changed files with 3 additions and 3 deletions
Show Stats Expand all files Collapse all files

2
wan/image2video.py
View File

@ -207,7 +207,7 @@ class WanI2V:
generator=seed_g, generator=seed_g,
device=self.device) device=self.device)
msk = torch.ones(1, 81, lat_h, lat_w, device=self.device) msk = torch.ones(1, F, lat_h, lat_w, device=self.device)
msk[:, 1:] = 0 msk[:, 1:] = 0
msk = torch.concat([ msk = torch.concat([
torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:] torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:]

4
wan/vace.py
View File

@ -352,7 +352,7 @@ class WanVace(WanT2V):
if n_prompt == "": if n_prompt == "":
n_prompt = self.sample_neg_prompt n_prompt = self.sample_neg_prompt
seed = seed if seed >= 0 else random.randint(0, sys.maxsize) seed = seed if seed >= 0 else random.randint(0, 1e7)
seed_g = torch.Generator(device=self.device) seed_g = torch.Generator(device=self.device)
seed_g.manual_seed(seed) seed_g.manual_seed(seed)
@ -649,7 +649,7 @@ class WanVaceMP(WanVace):
if n_prompt == "": if n_prompt == "":
n_prompt = sample_neg_prompt n_prompt = sample_neg_prompt
seed = seed if seed >= 0 else random.randint(0, sys.maxsize) seed = seed if seed >= 0 else random.randint(0, 1e7)
seed_g = torch.Generator(device=gpu) seed_g = torch.Generator(device=gpu)
seed_g.manual_seed(seed) seed_g.manual_seed(seed)