mfajcik commited on
Commit
ee26f4b
·
verified ·
1 Parent(s): 51b0d25

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ input_args = {
45
  import torch
46
  import transformers
47
 
48
- model_name = "BUT-FIT/CSTinyLLama-1.2B-SearchSummarizer"
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