ekshat commited on
Commit
47b5ca8
1 Parent(s): fed424d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  datasets:
3
- - b-mc2/sql-create-context
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 alpaca format described by Meta. The dataset is provided by "b-mc2/sql-create-context" present on Huggingface . We have used QLora, Bits&Bytes, Accelerate and Transformers Library to implement PEFT concept. We have fine-tuned this model based on pre-trained llama-2 7B model provided by 'NousResearch/Llama-2-7b-chat-hf'.
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 = "b-mc2/sql-create-context"**
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