Skip to content

GitLab

Commit 8421734f authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

Merge pull request #49 from ModelTC/dev_fix 

Fix
parents 2fb721ab 3fb9aa7e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
lightx2v/models/networks/wan/infer/transformer_infer.py
View file @ 8421734f
...@@ -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