Skip to content

GitLab

Commit 3fb9aa7e authored by gushiqiao's avatar gushiqiao
Browse files

Fix

parent 4fd83968
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
lightx2v/models/networks/wan/infer/transformer_infer.py
View file @ 3fb9aa7e
...@@ -50,6 +50,7 @@ class WanTransformerInfer: ...@@ -50,6 +50,7 @@ class WanTransformerInfer:
x = self.infer_block( x = self.infer_block(
self.weights_stream_mgr.active_weights[0], self.weights_stream_mgr.active_weights[0],
grid_sizes, grid_sizes,
embed,
x, x,
embed0, embed0,
seq_lens, seq_lens,
...@@ -119,6 +120,7 @@ class WanTransformerInfer: ...@@ -119,6 +120,7 @@ class WanTransformerInfer:
x = self.infer_block( x = self.infer_block(
weights.blocks[block_idx], weights.blocks[block_idx],
grid_sizes, grid_sizes,
embed,
x, x,
embed0, embed0,
seq_lens, seq_lens,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment