Update README.md
Browse files
README.md
CHANGED
@@ -90,6 +90,7 @@ prompt = f"Instruct: generate a SQL query.\n{prompt}\n\n" # for dataset b-mc2/sq
|
|
90 |
outputs = pipe(prompt, max_new_tokens=1024, do_sample=True, temperature=0.9, top_k=50, top_p=0.1, eos_token_id=pipe.tokenizer.eos_token_id, pad_token_id=pipe.tokenizer.eos_token_id)
|
91 |
|
92 |
print('Question: %s'%prompt)
|
|
|
93 |
print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
|
94 |
|
95 |
|
|
|
90 |
outputs = pipe(prompt, max_new_tokens=1024, do_sample=True, temperature=0.9, top_k=50, top_p=0.1, eos_token_id=pipe.tokenizer.eos_token_id, pad_token_id=pipe.tokenizer.eos_token_id)
|
91 |
|
92 |
print('Question: %s'%prompt)
|
93 |
+
|
94 |
print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
|
95 |
|
96 |
|