OpenNLPLab commited on
Commit
60b0f0a
·
verified ·
1 Parent(s): 60c48a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -61,8 +61,8 @@ This official repository unveils the TransNormerLLM3 model along with its open-s
61
  ```python
62
  from transformers import AutoModelForCausalLM, AutoTokenizer
63
 
64
- tokenizer = AutoTokenizer.from_pretrained("OpenNLPLab/TransNormerLLM3-15B-Intermediate-Checkpoints", revision='step26000-100Btokens', trust_remote_code=True)
65
- model = AutoModelForCausalLM.from_pretrained("OpenNLPLab/TransNormerLLM3-15B-Intermediate-Checkpoints", torch_dtype=torch.bfloat16, revision='step26000-100Btokens', device_map="auto", trust_remote_code=True)
66
  ```
67
 
68
  # Benchmark Results
 
61
  ```python
62
  from transformers import AutoModelForCausalLM, AutoTokenizer
63
 
64
+ tokenizer = AutoTokenizer.from_pretrained("OpenNLPLab/TransNormerLLM3-15B-Intermediate-Checkpoints", revision='step170000-650Btokens', trust_remote_code=True)
65
+ model = AutoModelForCausalLM.from_pretrained("OpenNLPLab/TransNormerLLM3-15B-Intermediate-Checkpoints", torch_dtype=torch.bfloat16, revision='step170000-650Btokens', device_map="auto", trust_remote_code=True)
66
  ```
67
 
68
  # Benchmark Results