GGmorello commited on
Commit
5413714
·
verified ·
1 Parent(s): b09d9f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ model = LlamaForCausalLM.from_pretrained(
19
 
20
 
21
  MAX_SEQ_LEN = 4096
22
- tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
23
- model.pad_token = tokenizer.pad_token = tokenizer.unk_token
24
 
25
 
26
 
 
19
 
20
 
21
  MAX_SEQ_LEN = 4096
22
+ tokenizer = AutoTokenizer.from_pretrained("GGmorello/FLAMES-100k")
23
+ model.config.pad_token = tokenizer.pad_token = tokenizer.unk_token
24
 
25
 
26