Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ Get started generating text with `Stable Zephyr 3B` by using the following code
|
|
| 31 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 32 |
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stable-zephyr-3b-dpo")
|
| 33 |
model = AutoModelForCausalLM.from_pretrained(
|
| 34 |
-
"
|
| 35 |
trust_remote_code=True,
|
| 36 |
torch_dtype="auto",
|
| 37 |
)
|
|
|
|
| 31 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 32 |
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stable-zephyr-3b-dpo")
|
| 33 |
model = AutoModelForCausalLM.from_pretrained(
|
| 34 |
+
"stable-zephyr-3b",
|
| 35 |
trust_remote_code=True,
|
| 36 |
torch_dtype="auto",
|
| 37 |
)
|