Spaces:
Runtime error
Runtime error
peter szemraj
commited on
Commit
•
bd355c2
1
Parent(s):
a9d9264
update params
Browse files- 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=
|
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 |
)
|