(self, c, x, **kwargs)
| 31 | nn.init.zeros_(self.after_proj.bias) |
| 32 | |
| 33 | def forward(self, c, x, **kwargs): |
| 34 | if self.block_id == 0: |
| 35 | c = self.before_proj(c) + x |
| 36 | |
| 37 | c = super().forward(c, **kwargs) |
| 38 | c_skip = self.after_proj(c) |
| 39 | return c, c_skip |
| 40 | |
| 41 | |
| 42 | class BaseWanAttentionBlock(WanAttentionBlock): |