Commit
•
c74edcc
1
Parent(s):
fce70f2
Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ tags:
|
|
14 |
```
|
15 |
import torch
|
16 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
17 |
-
base_model = 'bigdefence/Llama-3.1-8B-bigdefence
|
18 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
19 |
|
20 |
tokenizer = AutoTokenizer.from_pretrained(base_model)
|
|
|
14 |
```
|
15 |
import torch
|
16 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
17 |
+
base_model = 'bigdefence/Llama-3.1-8B-Ko-bigdefence'
|
18 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
19 |
|
20 |
tokenizer = AutoTokenizer.from_pretrained(base_model)
|