Update README.md
Browse files
README.md
CHANGED
@@ -86,7 +86,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
86 |
import torch
|
87 |
from rubra_tools import preprocess_input, postprocess_output
|
88 |
|
89 |
-
model_id = "rubra-ai/
|
90 |
|
91 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
92 |
model = AutoModelForCausalLM.from_pretrained(
|
@@ -275,7 +275,7 @@ Users should ensure that the deployment of this model adheres to ethical guideli
|
|
275 |
|
276 |
## Acknowledgements
|
277 |
|
278 |
-
We would like to thank Mistral for the model
|
279 |
|
280 |
## Contact Information
|
281 |
|
|
|
86 |
import torch
|
87 |
from rubra_tools import preprocess_input, postprocess_output
|
88 |
|
89 |
+
model_id = "rubra-ai/Mistral-7B-Instruct-v0.2"
|
90 |
|
91 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
92 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
275 |
|
276 |
## Acknowledgements
|
277 |
|
278 |
+
We would like to thank Mistral for the model.
|
279 |
|
280 |
## Contact Information
|
281 |
|