(self)
| 379 | self.out_proj = apply_wd(zero_init(Linear(d_model, d_model, bias=False))) |
| 380 | |
| 381 | def extra_repr(self): |
| 382 | return f"d_head={self.d_head}," |
| 383 | |
| 384 | def forward(self, x, pos, cond): |
| 385 | skip = x |
nothing calls this directly
no outgoing calls
no test coverage detected