Update README.md
Browse files
README.md
CHANGED
|
@@ -39,8 +39,8 @@ def promptify(prompt, pred, gold):
|
|
| 39 |
|
| 40 |
return prompt_template
|
| 41 |
|
| 42 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 43 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 44 |
|
| 45 |
prompt, pred, gold = '1、约翰喜欢看电影，玛丽也喜欢。\n2、约翰也喜欢看足球比赛。\n请问以上两句话是否是一个意思？', "不一样", "不一样"
|
| 46 |
|
|
|
|
| 39 |
|
| 40 |
return prompt_template
|
| 41 |
|
| 42 |
+
model = AutoModelForCausalLM.from_pretrained("FlagEval/flageval_judgemodel", torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, attn_implementation="flash_attention_2").cuda()
|
| 43 |
+
tokenizer = AutoTokenizer.from_pretrained("FlagEval/flageval_judgemodel")
|
| 44 |
|
| 45 |
prompt, pred, gold = '1、约翰喜欢看电影，玛丽也喜欢。\n2、约翰也喜欢看足球比赛。\n请问以上两句话是否是一个意思？', "不一样", "不一样"
|
| 46 |
|