lv12
/

bert_base_uncased_embedding_moe

Model card Files Files and versions Community

lv12 commited on Mar 13

Commit

82b10cf

·

verified ·

1 Parent(s): ee9e833

Uploading model.pt

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -116,4 +116,4 @@ class EmbeddingMoE(PreTrainedModel):
             return self.forward(input_ids, attention_mask)
-AutoModel.register(EmbeddingMoE)


116	return self.forward(input_ids, attention_mask)
117
118
119	+ AutoModel.register("embedding_moe", EmbeddingMoE)