Uploading model.pt
Browse files
model.py
CHANGED
|
@@ -115,6 +115,3 @@ class EmbeddingMoE(PreTrainedModel):
|
|
| 115 |
"""Helper method to get the embedding for a single sentence"""
|
| 116 |
with torch.no_grad():
|
| 117 |
return self.forward(input_ids, attention_mask)
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
AutoModel.register("embedding_moe", EmbeddingMoE)
|
|
|
|
| 115 |
"""Helper method to get the embedding for a single sentence"""
|
| 116 |
with torch.no_grad():
|
| 117 |
return self.forward(input_ids, attention_mask)
|
|
|
|
|
|
|
|
|