(self, image_embeds)
| 203 | ) |
| 204 | |
| 205 | def forward(self, image_embeds): |
| 206 | clip_extra_context_tokens = self.proj(image_embeds) |
| 207 | return clip_extra_context_tokens |
| 208 | |
| 209 | class MLPProjModelFaceId(nn.Module): |
| 210 | def __init__(self, cross_attention_dim=768, id_embeddings_dim=512, num_tokens=4): |
nothing calls this directly
no outgoing calls
no test coverage detected