Zhengping
/

conditional-probability-regression

Text Generation

text-generation-inference

Model card Files Files and versions Community

Zhengping commited on Apr 21

Commit

2f3d4d8

·

verified ·

1 Parent(s): 122d947

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -62,12 +62,12 @@ from src.chat_templates import UNLITemplate
 model = transformers.AutoModelForCausalLM.from_pretrained(
-    "ckpt/merged-dp3",
     torch_dtype="auto",
     attn_implementation="flash_attention_2",
 )
 tokenizer = transformers.AutoTokenizer.from_pretrained(
-    "ckpt/merged-dp3",
 )
 rank_dict = SingleLabelRankDict.from_tokenizer(tokenizer)

 model = transformers.AutoModelForCausalLM.from_pretrained(
+    "Zhengping/conditional-probability-regression",
     torch_dtype="auto",
     attn_implementation="flash_attention_2",
 )
 tokenizer = transformers.AutoTokenizer.from_pretrained(
+    "Zhengping/conditional-probability-regression",
 )
 rank_dict = SingleLabelRankDict.from_tokenizer(tokenizer)