damerajee commited on
Commit
760a46d
·
verified ·
1 Parent(s): 1b47c45

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +2 -0
modeling_gpt2vision.py CHANGED
@@ -101,6 +101,8 @@ class GPT2Vision(PreTrainedModel):
101
  pad_token_id=self.tokenizer.eos_token_id,
102
  eos_token_id=self.tokenizer.eos_token_id,
103
  max_new_tokens=max_new_tokens,
 
 
104
  **kwargs
105
  )
106
  output = self.tokenizer.decode(output_sequences[0], skip_special_tokens=True)
 
101
  pad_token_id=self.tokenizer.eos_token_id,
102
  eos_token_id=self.tokenizer.eos_token_id,
103
  max_new_tokens=max_new_tokens,
104
+ repetition_penalty=2.0,
105
+
106
  **kwargs
107
  )
108
  output = self.tokenizer.decode(output_sequences[0], skip_special_tokens=True)