mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-12-18 21:22:11 +00:00
Compare commits
3 Commits
1aab25d613
...
8e6dbce161
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8e6dbce161 | ||
|
|
841fe5237b | ||
|
|
e649b328e9 |
@ -207,7 +207,7 @@ class WanI2V:
|
||||
generator=seed_g,
|
||||
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 = torch.concat([
|
||||
torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:]
|
||||
|
||||
@ -352,7 +352,7 @@ class WanVace(WanT2V):
|
||||
|
||||
if n_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.manual_seed(seed)
|
||||
|
||||
@ -649,7 +649,7 @@ class WanVaceMP(WanVace):
|
||||
|
||||
if n_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.manual_seed(seed)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user