TwT-6 commited on
Commit
b9fa8ea
1 Parent(s): 88679c6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -14,11 +14,11 @@ import torch
14
 
15
 
16
  model = AutoModelForCausalLM.from_pretrained(
17
- 'TwT-6/open_llm_leaderboard_demo',
18
  attn_implementation="flash_attention_2",
19
  trust_remote_code=True, torch_dtype=torch.bfloat16, device_map="auto").eval()
20
 
21
- tokenizer = AutoTokenizer.from_pretrained('TwT-6/open_llm_leaderboard_demo', trust_remote_code=True)
22
 
23
  inputs = '你好'
24
 
 
14
 
15
 
16
  model = AutoModelForCausalLM.from_pretrained(
17
+ 'TwT-6/cr-model',
18
  attn_implementation="flash_attention_2",
19
  trust_remote_code=True, torch_dtype=torch.bfloat16, device_map="auto").eval()
20
 
21
+ tokenizer = AutoTokenizer.from_pretrained('TwT-6/cr-model', trust_remote_code=True)
22
 
23
  inputs = '你好'
24