mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2026-02-06 10:47:46 +00:00
Compare commits
3 Commits
f51c6a0731
...
c47fc07b9e
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c47fc07b9e | ||
|
|
841fe5237b | ||
|
|
e8820d0ac8 |
@ -205,4 +205,8 @@ if __name__ == '__main__':
|
||||
print("done", flush=True)
|
||||
|
||||
demo = gradio_interface()
|
||||
if "WAN_USER" in os.environ and "WAN_PWD" in os.environ:
|
||||
demo.launch(server_name="0.0.0.0", share=False, server_port=7860,
|
||||
auth=(os.environ["WAN_USER"], os.environ["WAN_PWD"]))
|
||||
else:
|
||||
demo.launch(server_name="0.0.0.0", share=False, server_port=7860)
|
||||
|
||||
@ -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:]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user