github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

typo

pull/479/head
x54-729 2023-11-07 20:23:14 +08:00
parent 08b6567ab5
commit 7b831a6776
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -157,7 +157,7 @@ We conducted a comprehensive evaluation of InternLM using the open-source evalua
To load the InternLM 7B Chat model using Transformers, use the following code:
```python
import
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("internlm/internlm-chat-7b-v1_1", trust_remote_code=True)
# Set `torch_dtype=torch.float16` to load model in float16, otherwise it will be loaded as float32 and might cause OOM Error.