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
f185da14
Commit
f185da14
authored
Aug 25, 2025
by
gushiqiao
Committed by
GitHub
Aug 25, 2025
Browse files
[Fix] fix vace bug (#242)
* [Fix] fix bug * Fix bug * Update run_wan_vace.sh
parent
3488b187
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
6 deletions
+3
-6
lightx2v/models/networks/wan/infer/module_io.py
lightx2v/models/networks/wan/infer/module_io.py
+2
-5
lightx2v/models/runners/wan/wan_vace_runner.py
lightx2v/models/runners/wan/wan_vace_runner.py
+1
-1
No files found.
lightx2v/models/networks/wan/infer/module_io.py
View file @
f185da14
from
dataclasses
import
dataclass
from
dataclasses
import
dataclass
,
field
from
typing
import
Any
,
Dict
from
typing
import
Any
,
Dict
import
torch
import
torch
...
@@ -6,7 +6,6 @@ import torch
...
@@ -6,7 +6,6 @@ import torch
@
dataclass
@
dataclass
class
WanPreInferModuleOutput
:
class
WanPreInferModuleOutput
:
# wan base model
embed
:
torch
.
Tensor
embed
:
torch
.
Tensor
grid_sizes
:
torch
.
Tensor
grid_sizes
:
torch
.
Tensor
x
:
torch
.
Tensor
x
:
torch
.
Tensor
...
@@ -14,6 +13,4 @@ class WanPreInferModuleOutput:
...
@@ -14,6 +13,4 @@ class WanPreInferModuleOutput:
seq_lens
:
torch
.
Tensor
seq_lens
:
torch
.
Tensor
freqs
:
torch
.
Tensor
freqs
:
torch
.
Tensor
context
:
torch
.
Tensor
context
:
torch
.
Tensor
adapter_output
:
Dict
[
str
,
Any
]
=
field
(
default_factory
=
dict
)
# wan adapter model
adapter_output
:
Dict
[
str
,
Any
]
=
None
lightx2v/models/runners/wan/wan_vace_runner.py
View file @
f185da14
...
@@ -153,7 +153,7 @@ class WanVaceRunner(WanRunner):
...
@@ -153,7 +153,7 @@ class WanVaceRunner(WanRunner):
self
.
config
.
target_shape
=
target_shape
self
.
config
.
target_shape
=
target_shape
@
ProfilingContext
(
"Run VAE Decoder"
)
@
ProfilingContext
(
"Run VAE Decoder"
)
def
_
run_vae_decoder
_local
(
self
,
latents
,
generator
):
def
run_vae_decoder
(
self
,
latents
,
generator
):
if
self
.
config
.
get
(
"lazy_load"
,
False
)
or
self
.
config
.
get
(
"unload_modules"
,
False
):
if
self
.
config
.
get
(
"lazy_load"
,
False
)
or
self
.
config
.
get
(
"unload_modules"
,
False
):
self
.
vae_decoder
=
self
.
load_vae_decoder
()
self
.
vae_decoder
=
self
.
load_vae_decoder
()
...
...
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