nroggendorff
commited on
Commit
•
d9b7328
1
Parent(s):
78f7207
Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ bnb_config = BitsAndBytesConfig(
|
|
41 |
bnb_4bit_compute_dtype=torch.bfloat16
|
42 |
)
|
43 |
|
44 |
-
model_id = "nroggendorff/eap"
|
45 |
|
46 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
47 |
model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
|
|
|
41 |
bnb_4bit_compute_dtype=torch.bfloat16
|
42 |
)
|
43 |
|
44 |
+
model_id = "nroggendorff/mistral-eap"
|
45 |
|
46 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
47 |
model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
|