Update README.md
Browse files
    	
        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=' | 
| 65 | 
            -
            model = AutoModelForCausalLM.from_pretrained("OpenNLPLab/TransNormerLLM3-15B-Intermediate-Checkpoints", torch_dtype=torch.bfloat16, revision=' | 
| 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
         | 
