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

Update README.md

pull/7/head
Shuo Zhang 2023-07-06 16:37:59 +08:00 committed by GitHub
parent ce02dcf54f
commit c4d1d96c06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -144,7 +144,7 @@ python convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer
After conversion, it can be loaded as transformers by the following code After conversion, it can be loaded as transformers by the following code
```python ```python
>>> from transformers import AutoTokenizer, AutoModel >>> from transformers import AutoTokenizer, AutoModel
>>> model = AutoModel.from_pretrained("hf_ckpt/", trust_remote_code=True, device='cuda') >>> model = AutoModel.from_pretrained("hf_ckpt/", trust_remote_code=True).cuda()
``` ```