Commit
•
2674262
1
Parent(s):
883b280
Update README.md
Browse files
README.md
CHANGED
@@ -28,7 +28,7 @@ nf4_config = BitsAndBytesConfig(
|
|
28 |
bnb_4bit_compute_dtype=getattr(torch, TORCH_DTYPE)
|
29 |
)
|
30 |
|
31 |
-
tokenizer = AutoTokenizer.from_pretrained('mesolitica/malaysian-mistral-3B-4096'
|
32 |
model = AutoModelForCausalLM.from_pretrained(
|
33 |
'mesolitica/malaysian-mistral-3B-4096',
|
34 |
use_flash_attention_2 = True,
|
|
|
28 |
bnb_4bit_compute_dtype=getattr(torch, TORCH_DTYPE)
|
29 |
)
|
30 |
|
31 |
+
tokenizer = AutoTokenizer.from_pretrained('mesolitica/malaysian-mistral-3B-4096')
|
32 |
model = AutoModelForCausalLM.from_pretrained(
|
33 |
'mesolitica/malaysian-mistral-3B-4096',
|
34 |
use_flash_attention_2 = True,
|