damerajee commited on
Commit
8c80497
·
verified ·
1 Parent(s): c489aa6

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +2 -0
modeling_gpt2vision.py CHANGED
@@ -167,6 +167,8 @@ class GPT2Vision(PreTrainedModel):
167
  pad_token_id=self.tokenizer.eos_token_id,
168
  eos_token_id=self.tokenizer.eos_token_id,
169
  max_new_tokens=max_new_tokens,
 
 
170
  **kwargs
171
  )
172
 
 
167
  pad_token_id=self.tokenizer.eos_token_id,
168
  eos_token_id=self.tokenizer.eos_token_id,
169
  max_new_tokens=max_new_tokens,
170
+ repetition_penalty=1.0,
171
+
172
  **kwargs
173
  )
174