Update README.md
Browse files
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
datasets:
|
3 |
-
-
|
4 |
language:
|
5 |
- en
|
6 |
library_name: transformers
|
@@ -11,7 +11,7 @@ tags:
|
|
11 |
- text2sql
|
12 |
---
|
13 |
# Model Description
|
14 |
-
Our Model is fine tuned on Llama-2 7B model on text-2-sql Dataset on
|
15 |
|
16 |
|
17 |
# Inference
|
@@ -43,7 +43,7 @@ print(result[0]['generated_text'])
|
|
43 |
# Model Information
|
44 |
- **model_name = "NousResearch/Llama-2-7b-chat-hf"**
|
45 |
|
46 |
-
- **dataset_name = "
|
47 |
|
48 |
|
49 |
# QLoRA parameters
|
|
|
1 |
---
|
2 |
datasets:
|
3 |
+
- ekshat/text-2-sql-with-context
|
4 |
language:
|
5 |
- en
|
6 |
library_name: transformers
|
|
|
11 |
- text2sql
|
12 |
---
|
13 |
# Model Description
|
14 |
+
Our Model is fine tuned on Llama-2 7B model on text-2-sql Dataset on Alpaca format described by Stanford. We have used QLora, Bits&Bytes, Accelerate and Transformers Library to implement PEFT concept.
|
15 |
|
16 |
|
17 |
# Inference
|
|
|
43 |
# Model Information
|
44 |
- **model_name = "NousResearch/Llama-2-7b-chat-hf"**
|
45 |
|
46 |
+
- **dataset_name = "ekshat/text-2-sql-with-context"**
|
47 |
|
48 |
|
49 |
# QLoRA parameters
|