Tuchuanhuhuhu commited on
Commit
a66696f
·
1 Parent(s): 3f11a48

fix invalid_request_error in GPT models

Browse files
Files changed (1) hide show
  1. modules/models/OpenAIVision.py +6 -3
modules/models/OpenAIVision.py CHANGED
@@ -171,11 +171,14 @@ class OpenAIVisionClient(BaseLLMModel):
171
  "top_p": self.top_p,
172
  "n": self.n_choices,
173
  "stream": stream,
174
- "presence_penalty": self.presence_penalty,
175
- "frequency_penalty": self.frequency_penalty,
176
- "max_tokens": self.max_generation_token
177
  }
178
 
 
 
 
 
 
 
179
  if self.stop_sequence:
180
  payload["stop"] = self.stop_sequence
181
  if self.logit_bias is not None:
 
171
  "top_p": self.top_p,
172
  "n": self.n_choices,
173
  "stream": stream,
 
 
 
174
  }
175
 
176
+ if self.max_generation_token:
177
+ payload["max_tokens"] = self.max_generation_token
178
+ if self.presence_penalty:
179
+ payload["presence_penalty"] = self.presence_penalty
180
+ if self.frequency_penalty:
181
+ payload["frequency_penalty"] = self.frequency_penalty
182
  if self.stop_sequence:
183
  payload["stop"] = self.stop_sequence
184
  if self.logit_bias is not None: