bug

labdao · Apr 14, 2023 · 2fb307a · 2fb307a
1 parent 826b880
commit 2fb307a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/protbert/app.py b/protbert/app.py
@@ -137,7 +137,7 @@ def main(
     print(f"Using model {huggingface_model_name}")
 
     # Load models and tokenizer
-    tokenizer = AutoTokenizer.from_pretrained(model_name, do_lower_case=False)
+    tokenizer = AutoTokenizer.from_pretrained(huggingface_model_name, do_lower_case=False)
     masked_model = AutoModelForMaskedLM.from_pretrained(huggingface_model_name)
     generator_model = AutoModelForCausalLM.from_pretrained(huggingface_model_name)
     plain_model = AutoModel.from_pretrained(huggingface_model_name)