yuchenglu commited on
Commit
0fb7015
1 Parent(s): a8043fd

Update README.md: fix the typos for model loading example

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -40,8 +40,8 @@ Alternatively, you can load the model directly from the Hugging Face model hub u
40
  ```python
41
  from transformers import AutoTokenizer, AutoModelForCausalLM
42
 
43
- tokenizer = AutoTokenizer.from_pretrained("togethercomputer/LLaMA-2-7B-32K")
44
- model = AutoModelForCausalLM.from_pretrained("togethercomputer/LLaMA-2-7B-32K", trust_remote_code=True, torch_dtype=torch.float16)
45
  ```
46
 
47
  The model is also hosted on [Together Playground](https://api.together.xyz/playground). You can simply play with the model by using prompt formatted by
 
40
  ```python
41
  from transformers import AutoTokenizer, AutoModelForCausalLM
42
 
43
+ tokenizer = AutoTokenizer.from_pretrained("togethercomputer/LLaMA-2-7B-32K-Chat")
44
+ model = AutoModelForCausalLM.from_pretrained("togethercomputer/LLaMA-2-7B-32K-Chat", trust_remote_code=True, torch_dtype=torch.float16)
45
  ```
46
 
47
  The model is also hosted on [Together Playground](https://api.together.xyz/playground). You can simply play with the model by using prompt formatted by