Commit
•
53da47b
1
Parent(s):
3d12574
Update README.md
Browse files
README.md
CHANGED
@@ -159,7 +159,7 @@ from transformers import pipeline
|
|
159 |
|
160 |
pipe = pipeline(
|
161 |
"text-generation",
|
162 |
-
model="HuggingFaceH4/zephyr-7b-gemma",
|
163 |
device_map="auto",
|
164 |
torch_dtype=torch.bfloat16,
|
165 |
)
|
|
|
159 |
|
160 |
pipe = pipeline(
|
161 |
"text-generation",
|
162 |
+
model="HuggingFaceH4/zephyr-7b-gemma-v0.1",
|
163 |
device_map="auto",
|
164 |
torch_dtype=torch.bfloat16,
|
165 |
)
|