Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ from transformers import pipeline
|
|
| 35 |
|
| 36 |
extractor = pipeline(
|
| 37 |
"text-generation",
|
| 38 |
-
model="
|
| 39 |
model_kwargs=dict(load_in_4bit=True, torch_dtype="auto"),
|
| 40 |
)
|
| 41 |
|
|
|
|
| 35 |
|
| 36 |
extractor = pipeline(
|
| 37 |
"text-generation",
|
| 38 |
+
model="boods/mistral-location-extractor-4bit",
|
| 39 |
model_kwargs=dict(load_in_4bit=True, torch_dtype="auto"),
|
| 40 |
)
|
| 41 |
|