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] = []
|