(self, x: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
| 324 | return self.resblocks[0].mlp.c_fc.weight.device |
| 325 | |
| 326 | def forward(self, x: torch.Tensor, attn_mask: Optional[torch.Tensor] = None): |
| 327 | for r in self.resblocks: |
| 328 | x = r(x, attn_mask=attn_mask) |
| 329 | return x |
| 330 | |
| 331 | |
| 332 | class VisionTransformer(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected