| 77 | self.pipeline[k] = v |
| 78 | |
| 79 | class WanVideoModelConfig: |
| 80 | def __init__(self, latent_format=latent_format): |
| 81 | self.unet_config = {} |
| 82 | self.unet_extra_config = {} |
| 83 | self.latent_format = latent_format |
| 84 | |
| 85 | def filter_state_dict_by_blocks(state_dict, blocks_mapping, layer_filter=[]): |
| 86 | filtered_dict = {} |