(self)
| 42 | self.spell_embeddings.weight.data[i] = target_embedding |
| 43 | |
| 44 | def forward(self): |
| 45 | prompt_embeds = self.spell_embeddings.weight.unsqueeze(0) |
| 46 | if self.spell_func == "lstm": |
| 47 | prompt_embeds = self.lstm_head(prompt_embeds)[0] |
| 48 | if self.spell_func == "lstm" or self.spell_func == "mlp": |
| 49 | prompt_embeds = self.mlp_head(prompt_embeds) |
| 50 | return prompt_embeds |
nothing calls this directly
no outgoing calls
no test coverage detected