(self, latents, x)
| 98 | nn.init.constant_(self.proj_out.bias, 0) |
| 99 | |
| 100 | def forward(self, latents, x): |
| 101 | x = self.proj_in(x) |
| 102 | for attn, ff in self.layers: |
| 103 | latents = attn(x, latents) + latents |
| 104 | latents = ff(latents) + latents |
| 105 | latents = self.proj_out(latents) |
| 106 | return self.norm_out(latents) |
| 107 | |
| 108 | |
| 109 | class FusionFaceId(ModelMixin): |
nothing calls this directly
no outgoing calls
no test coverage detected