Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,7 @@ Make sure to update your transformers installation via pip install --upgrade tra
|
|
| 33 |
import transformers
|
| 34 |
import torch
|
| 35 |
|
| 36 |
-
model_id = "SemanticAlignment/
|
| 37 |
|
| 38 |
pipeline = transformers.pipeline(
|
| 39 |
"text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
|
|
|
|
| 33 |
import transformers
|
| 34 |
import torch
|
| 35 |
|
| 36 |
+
model_id = "SemanticAlignment/Mistral-v0.1-Italian-FVT"
|
| 37 |
|
| 38 |
pipeline = transformers.pipeline(
|
| 39 |
"text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
|