Tonic commited on
Commit
6282503
·
1 Parent(s): 7da94d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,9 +10,9 @@ peft_config = PeftConfig(
10
  max_length=500,
11
  use_cache=True,
12
  early_stopping=False,
13
- bos_token_id=model.config.bos_token_id,
14
- eos_token_id=model.config.eos_token_id,
15
- pad_token_id=model.config.eos_token_id,
16
  temperature=0.4,
17
  do_sample=True
18
  )
 
10
  max_length=500,
11
  use_cache=True,
12
  early_stopping=False,
13
+ bos_token_id=peft_model.config.bos_token_id,
14
+ eos_token_id=peft_model.config.eos_token_id,
15
+ pad_token_id=peft_model.config.eos_token_id,
16
  temperature=0.4,
17
  do_sample=True
18
  )