neofung
/

LdIR-Qwen2-reranker-1.5B

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

neofung commited on Aug 13

Commit

4cbcfab

•

1 Parent(s): cb971c7

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

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

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