indiejoseph
commited on
Commit
•
fec6e45
1
Parent(s):
f2a4dad
Update README.md
Browse files
README.md
CHANGED
@@ -21,8 +21,8 @@ This fine tuned model is intended to evaluate the imapct of Simplified Chinese i
|
|
21 |
```python
|
22 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
23 |
|
24 |
-
model = AutoModelForCausalLM.from_pretrained("cantonese-llama-2-7b-oasst-v1
|
25 |
-
tokenizer = AutoTokenizer.from_pretrained("cantonese-llama-2-7b-oasst-v1
|
26 |
|
27 |
template = """A chat between a curious user and an artificial intelligence assistant.
|
28 |
The assistant gives helpful, detailed, and polite answers to the user's questions.
|
|
|
21 |
```python
|
22 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
23 |
|
24 |
+
model = AutoModelForCausalLM.from_pretrained("indiejoseph/cantonese-llama-2-7b-oasst-v1", device_map="auto")
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained("indiejoseph/cantonese-llama-2-7b-oasst-v1")
|
26 |
|
27 |
template = """A chat between a curious user and an artificial intelligence assistant.
|
28 |
The assistant gives helpful, detailed, and polite answers to the user's questions.
|