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