Spaces:
Sleeping
Sleeping
Commit
·
36de391
1
Parent(s):
9f7caae
debug
Browse files
app.py
CHANGED
@@ -170,10 +170,9 @@ def get_response(params):
|
|
170 |
# RuntimeError: The size of tensor a (2) must match the size of tensor b (3) at non-singleton dimension 0
|
171 |
generate_kwargs = dict(
|
172 |
inputs=input_ids,
|
173 |
-
do_sample=
|
174 |
-
|
175 |
-
|
176 |
-
max_new_tokens=max_new_tokens,
|
177 |
pad_token_id=tokenizer.pad_token_id,
|
178 |
streamer=streamer,
|
179 |
# stopping_criteria=[stopping_criteria],
|
|
|
170 |
# RuntimeError: The size of tensor a (2) must match the size of tensor b (3) at non-singleton dimension 0
|
171 |
generate_kwargs = dict(
|
172 |
inputs=input_ids,
|
173 |
+
do_sample=False,
|
174 |
+
top_p=1.0,
|
175 |
+
max_new_tokens=512,
|
|
|
176 |
pad_token_id=tokenizer.pad_token_id,
|
177 |
streamer=streamer,
|
178 |
# stopping_criteria=[stopping_criteria],
|