nroggendorff commited on
Commit
c979c0c
1 Parent(s): 47ec969

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -30,7 +30,7 @@ To use the Mayo LLM, you can load the model using the Hugging Face Transformers
30
  ```python
31
  from transformers import pipeline
32
 
33
- pipe = pipeline("text-generation", model="nroggendorff/mayo")
34
 
35
  question = "What color is the sky?"
36
  conv = [{"role": "user", "content": question}]
@@ -52,7 +52,7 @@ bnb_config = BitsAndBytesConfig(
52
  bnb_4bit_compute_dtype=torch.bfloat16
53
  )
54
 
55
- model_id = "nroggendorff/mayo"
56
 
57
  tokenizer = AutoTokenizer.from_pretrained(model_id)
58
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
 
30
  ```python
31
  from transformers import pipeline
32
 
33
+ pipe = pipeline("text-generation", model="nroggendorff/vegetarian-mayo")
34
 
35
  question = "What color is the sky?"
36
  conv = [{"role": "user", "content": question}]
 
52
  bnb_4bit_compute_dtype=torch.bfloat16
53
  )
54
 
55
+ model_id = "nroggendorff/vegetarian-mayo"
56
 
57
  tokenizer = AutoTokenizer.from_pretrained(model_id)
58
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)