Spaces:
Sleeping
Sleeping
Update duckdb-nsql/eval/text_to_sql.py
Browse files
duckdb-nsql/eval/text_to_sql.py
CHANGED
|
@@ -151,7 +151,7 @@ def instruction_to_sql_list(
|
|
| 151 |
logprobs=5,
|
| 152 |
temperature=0.1,
|
| 153 |
do_sample=False,
|
| 154 |
-
stop_sequences=
|
| 155 |
)
|
| 156 |
|
| 157 |
ret: list[TextToSQLModelResponse] = []
|
|
|
|
| 151 |
logprobs=5,
|
| 152 |
temperature=0.1,
|
| 153 |
do_sample=False,
|
| 154 |
+
stop_sequences=stop_sequences or prompt_formatter.stop_sequences,
|
| 155 |
)
|
| 156 |
|
| 157 |
ret: list[TextToSQLModelResponse] = []
|