nroggendorff
commited on
Commit
•
c979c0c
1
Parent(s):
47ec969
Update README.md
Browse files
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)
|