Update README.md: fix the typos for model loading example
Browse files
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
|