Upload folder using huggingface_hub

README.md

@@ -16,7 +16,7 @@ import torch
 model_id = "gustoudu81/BerTeleo"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
-model = AutoModelForMaskedLM.from_pretrained(model_id, trust_remote_code=True)
+model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device).eval()

config.json

@@ -1,8 +1,5 @@
 {
   "alibi_starting_size": 512,
-  "architectures": [
-    "BertForMaskedLM"
-  ],
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
     "AutoConfig": "gustoudu81/DNABERT-2-117M-tritonfix--configuration_bert.BertConfig",