ekshat
/

Llama-2-7b-chat-finetune-for-text2sql

Text Generation

text-generation-inference

Model card Files Files and versions

ekshat commited on Sep 14, 2023

Commit

a395344

·

1 Parent(s): 3bbcca7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ Our Model is fine tuned on Llama-2 7B model on text-2-sql Dataset on alpaca form
 # Inference
 ```python
-!pip install transformers accelerate
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline

 # Inference
 ```python
+!pip install transformers accelerate xformers
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline