peter szemraj commited on
Commit
bd355c2
1 Parent(s): a9d9264

update params

Browse files
Files changed (1) hide show
  1. ai_single_response.py +2 -1
ai_single_response.py CHANGED
@@ -111,7 +111,7 @@ def query_gpt_model(
111
  speaker=None,
112
  responder=None,
113
  resp_length=128,
114
- resp_min=4,
115
  kparam=150,
116
  temp=0.75,
117
  top_p=0.65,
@@ -168,6 +168,7 @@ def query_gpt_model(
168
  return_as_list=True,
169
  use_cache=True,
170
  num_beams=beams,
 
171
  early_stopping=True,
172
  # no_repeat_ngram_size=2,
173
  )
 
111
  speaker=None,
112
  responder=None,
113
  resp_length=128,
114
+ resp_min=10,
115
  kparam=150,
116
  temp=0.75,
117
  top_p=0.65,
 
168
  return_as_list=True,
169
  use_cache=True,
170
  num_beams=beams,
171
+ num_return_sequences=1,
172
  early_stopping=True,
173
  # no_repeat_ngram_size=2,
174
  )