ekshat commited on
Commit
8e32c30
·
1 Parent(s): 47b5ca8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ Our Model is fine tuned on Llama-2 7B model on text-2-sql Dataset on Alpaca form
21
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
22
 
23
  tokenizer = AutoTokenizer.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql")
24
- # Load model directly
25
  model = AutoModelForCausalLM.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql", load_in_4bit=True)
26
 
27
  context = "CREATE TABLE head (name VARCHAR, born_state VARCHAR, age VARCHAR)"
 
21
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
22
 
23
  tokenizer = AutoTokenizer.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql")
24
+
25
  model = AutoModelForCausalLM.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql", load_in_4bit=True)
26
 
27
  context = "CREATE TABLE head (name VARCHAR, born_state VARCHAR, age VARCHAR)"