SambaCoder-nsql-llama-2-70b-text-to-SQL

Sleeping

bol20162021 commited on Mar 1, 2024

Commit

93087d7

verified ·

1 Parent(s): 2a98158

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ def generate_sql(question, schema):
         "inputs": [prompt],
         "params": {
             "do_sample": {"type": "bool", "value": "false"},
-            "max_tokens_to_generate": {"type": "int", "value": "1000"},
             "repetition_penalty": {"type": "float", "value": "1"},
             "temperature": {"type": "float", "value": "1"},
             "top_k": {"type": "int", "value": "50"},

         "inputs": [prompt],
         "params": {
             "do_sample": {"type": "bool", "value": "false"},
+            "max_tokens_to_generate": {"type": "int", "value": "2048"},
             "repetition_penalty": {"type": "float", "value": "1"},
             "temperature": {"type": "float", "value": "1"},
             "top_k": {"type": "int", "value": "50"},