abir-hr196 commited on
Commit
3311e08
·
1 Parent(s): db9e075

alpaca prompt

Browse files
Files changed (1) hide show
  1. 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"""### Instruction: {instruction}
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(