Spaces:
Sleeping
Sleeping
Commit
·
3311e08
1
Parent(s):
db9e075
alpaca prompt
Browse files- tinysql_model_demo.py +2 -5
tinysql_model_demo.py
CHANGED
|
@@ -36,11 +36,8 @@ def generate_sql(model_name, instruction, schema, max_length=256, temperature=0.
|
|
| 36 |
|
| 37 |
try:
|
| 38 |
tokenizer, model = load_model(model_name)
|
| 39 |
-
|
| 40 |
-
prompt = f"
|
| 41 |
-
### Context: {schema}
|
| 42 |
-
### Response:"""
|
| 43 |
-
|
| 44 |
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 45 |
|
| 46 |
outputs = model.generate(
|
|
|
|
| 36 |
|
| 37 |
try:
|
| 38 |
tokenizer, model = load_model(model_name)
|
| 39 |
+
|
| 40 |
+
prompt = f"### Instruction: {instruction} ### Context: {schema} ### Response:"
|
|
|
|
|
|
|
|
|
|
| 41 |
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 42 |
|
| 43 |
outputs = model.generate(
|