Update README.md
Browse files
README.md
CHANGED
@@ -16,9 +16,9 @@ base_model : [Ko-Llama3-Luxia-8B](https://huggingface.co/saltlux/Ko-Llama3-Luxia
|
|
16 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
17 |
import torch
|
18 |
|
19 |
-
tokenizer = AutoTokenizer.from_pretrained("MDDDDR/Ko-Luxia-8B-it-v0.
|
20 |
model = AutoModelForCausalLM.from_pretrained(
|
21 |
-
"MDDDDR/Ko-Luxia-8B-it-v0.
|
22 |
device_map="auto",
|
23 |
torch_dtype=torch.bfloat16
|
24 |
)
|
|
|
16 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
17 |
import torch
|
18 |
|
19 |
+
tokenizer = AutoTokenizer.from_pretrained("MDDDDR/Ko-Luxia-8B-it-v0.2")
|
20 |
model = AutoModelForCausalLM.from_pretrained(
|
21 |
+
"MDDDDR/Ko-Luxia-8B-it-v0.2",
|
22 |
device_map="auto",
|
23 |
torch_dtype=torch.bfloat16
|
24 |
)
|