update model.py to work

model.py

@@ -16,8 +16,7 @@ def load_embedder_and_tokenizer(name: str) -> Tuple[
         transformers.PreTrainedTokenizer
 ]:
     if name.startswith("nomic") or (name == "bert-base-uncased"):
-        config = transformers.AutoConfig.from_pretrained(name, trust_remote_code=True)
-        model = transformers.AutoModel.from_pretrained(name, config=config, trust_remote_code=True)
+        model = transformers.AutoModelForMaskedLM.from_pretrained(name, trust_remote_code=True).bert
         tokenizer = transformers.AutoTokenizer.from_pretrained(name)
     elif name in ["gtr-base", "gtr_base"]:
         model = transformers.AutoModel.from_pretrained(