sanjay920 commited on
Commit
e441ffb
1 Parent(s): 8f4e1f4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/Meta-Llama-3-8B-Instruct"
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 and LLaMA-Factory for training utilities.
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