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 |
)
|