Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ input_args = {
|
|
45 |
import torch
|
46 |
import transformers
|
47 |
|
48 |
-
model_name = "BUT-FIT/CSTinyLLama-1.2B-
|
49 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
|
50 |
config = transformers.AutoConfig.from_pretrained(model_name, trust_remote_code=True)
|
51 |
|
|
|
45 |
import torch
|
46 |
import transformers
|
47 |
|
48 |
+
model_name = "BUT-FIT/CSTinyLLama-1.2B-RAGsum"
|
49 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
|
50 |
config = transformers.AutoConfig.from_pretrained(model_name, trust_remote_code=True)
|
51 |
|