Update README.md

README.md

@@ -171,18 +171,18 @@ class FlagRerankerCustom:
         return all_scores
 
 tokenizer = transformers.AutoTokenizer.from_pretrained(
-    "neofung/LdIR-Qwen2-reranker-1.5B-large",
+    "neofung/LdIR-Qwen2-reranker-1.5B",
     padding_side="right",
 )
 
 config = Qwen2Config.from_pretrained(
-    "neofung/LdIR-Qwen2-reranker-1.5B-large",
+    "neofung/LdIR-Qwen2-reranker-1.5B",
     trust_remote_code=True,
     bf16=True,
 )
 
 model = Qwen2ForSequenceClassification.from_pretrained(
-    "neofung/LdIR-Qwen2-reranker-1.5B-large",
+    "neofung/LdIR-Qwen2-reranker-1.5B",
     config = config,
     trust_remote_code = True,
 )