[BlockInfo] Wan (#10845)

* block info

* animate

* tensor

* device

* revert
This commit is contained in:
Haoming
2025-12-16 09:59:16 +08:00
committed by GitHub
parent fc4af86068
commit ea2c117bc3
2 changed files with 21 additions and 3 deletions

View File

@@ -523,7 +523,10 @@ class AnimateWanModel(WanModel):
patches_replace = transformer_options.get("patches_replace", {})
blocks_replace = patches_replace.get("dit", {})
transformer_options["total_blocks"] = len(self.blocks)
transformer_options["block_type"] = "double"
for i, block in enumerate(self.blocks):
transformer_options["block_index"] = i
if ("double_block", i) in blocks_replace:
def block_wrap(args):
out = {}