Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Model Card for Fine-Tuned Mistral 7B for Text-to-SQL Generation
|
| 2 |
|
| 3 |
## Model Details
|
|
@@ -69,4 +81,4 @@ outputs = model.generate(
|
|
| 69 |
)
|
| 70 |
|
| 71 |
generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 72 |
-
print(generated_sql)
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
datasets:
|
| 4 |
+
- gretelai/synthetic_text_to_sql
|
| 5 |
+
language:
|
| 6 |
+
- en
|
| 7 |
+
base_model:
|
| 8 |
+
- mistralai/Mistral-7B-v0.1
|
| 9 |
+
pipeline_tag: text2text-generation
|
| 10 |
+
tags:
|
| 11 |
+
- text-to-sql
|
| 12 |
+
---
|
| 13 |
# Model Card for Fine-Tuned Mistral 7B for Text-to-SQL Generation
|
| 14 |
|
| 15 |
## Model Details
|
|
|
|
| 81 |
)
|
| 82 |
|
| 83 |
generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 84 |
+
print(generated_sql)
|