Update README.md
Browse files
README.md
CHANGED
@@ -47,19 +47,17 @@ Aihub datasets 활용하여서 제작함.
|
|
47 |
|
48 |
## Implmentation Code
|
49 |
```
|
50 |
-
### KO-Platypus
|
51 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
52 |
import torch
|
53 |
|
54 |
repo = "MarkrAI/RAG-KO-Mixtral-7Bx2-v1.1"
|
55 |
-
|
56 |
repo,
|
57 |
return_dict=True,
|
58 |
torch_dtype=torch.float16,
|
59 |
device_map='auto'
|
60 |
)
|
61 |
-
|
62 |
-
|
63 |
```
|
64 |
|
65 |
# Model Benchmark
|
|
|
47 |
|
48 |
## Implmentation Code
|
49 |
```
|
|
|
50 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
51 |
import torch
|
52 |
|
53 |
repo = "MarkrAI/RAG-KO-Mixtral-7Bx2-v1.1"
|
54 |
+
markrAI_RAG = AutoModelForCausalLM.from_pretrained(
|
55 |
repo,
|
56 |
return_dict=True,
|
57 |
torch_dtype=torch.float16,
|
58 |
device_map='auto'
|
59 |
)
|
60 |
+
markrAI_RAG_tokenizer = AutoTokenizer.from_pretrained(repo)
|
|
|
61 |
```
|
62 |
|
63 |
# Model Benchmark
|