(self, model, block_swap_args=None)
| 58 | CATEGORY = "WanVideoWrapper" |
| 59 | |
| 60 | def loadmodel(self, model, block_swap_args=None): |
| 61 | if block_swap_args is None: |
| 62 | return (model,) |
| 63 | patcher = model.clone() |
| 64 | if 'transformer_options' not in patcher.model_options: |
| 65 | patcher.model_options['transformer_options'] = {} |
| 66 | patcher.model_options["transformer_options"]["block_swap_args"] = block_swap_args |
| 67 | |
| 68 | return (patcher,) |
| 69 | |
| 70 | class WanVideoSetRadialAttention: |
| 71 | @classmethod |