Update README.md
Browse files
README.md
CHANGED
@@ -31,10 +31,10 @@ import torch
|
|
31 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
32 |
|
33 |
tokenizer = AutoTokenizer.from_pretrained(
|
34 |
-
"taoki/Swallow-MS-7b-v0.1-qlora-
|
35 |
)
|
36 |
model = AutoModelForCausalLM.from_pretrained(
|
37 |
-
"taoki/Swallow-MS-7b-v0.1-qlora-
|
38 |
)
|
39 |
|
40 |
if torch.cuda.is_available():
|
|
|
31 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
32 |
|
33 |
tokenizer = AutoTokenizer.from_pretrained(
|
34 |
+
"taoki/Swallow-MS-7b-v0.1-qlora-oasst1-jmultiwoz-dolly-amenokaku"
|
35 |
)
|
36 |
model = AutoModelForCausalLM.from_pretrained(
|
37 |
+
"taoki/Swallow-MS-7b-v0.1-qlora-oasst1-jmultiwoz-dolly-amenokaku"
|
38 |
)
|
39 |
|
40 |
if torch.cuda.is_available():
|