Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
xuwx1
LightX2V
Commits
9196a220
"examples/vscode:/vscode.git/clone" did not exist on "5ff25cdf5b1310e83d9e595142b39ae4d7b561e9"
Commit
9196a220
authored
Aug 14, 2025
by
helloyongyang
Browse files
remove mask in wan22 5B t2v
parent
b469676c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
6 deletions
+3
-6
lightx2v/models/schedulers/wan/scheduler.py
lightx2v/models/schedulers/wan/scheduler.py
+3
-6
No files found.
lightx2v/models/schedulers/wan/scheduler.py
View file @
9196a220
...
@@ -60,12 +60,9 @@ class WanScheduler(BaseScheduler):
...
@@ -60,12 +60,9 @@ class WanScheduler(BaseScheduler):
device
=
self
.
device
,
device
=
self
.
device
,
generator
=
self
.
generator
,
generator
=
self
.
generator
,
)
)
if
self
.
config
[
"model_cls"
]
==
"wan2.2"
:
if
self
.
config
[
"model_cls"
]
==
"wan2.2"
and
self
.
config
[
"task"
]
==
"i2v"
:
if
self
.
config
[
"task"
]
==
"t2v"
:
self
.
mask
=
masks_like
(
self
.
latents
,
zero
=
True
)
self
.
mask
=
masks_like
(
self
.
latents
,
zero
=
False
)
self
.
latents
=
(
1.0
-
self
.
mask
)
*
self
.
vae_encoder_out
+
self
.
mask
*
self
.
latents
elif
self
.
config
[
"task"
]
==
"i2v"
:
self
.
mask
=
masks_like
(
self
.
latents
,
zero
=
True
)
self
.
latents
=
(
1.0
-
self
.
mask
)
*
self
.
vae_encoder_out
+
self
.
mask
*
self
.
latents
def
set_timesteps
(
def
set_timesteps
(
self
,
self
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment