support-pvelocity commited on
Commit
53e49aa
1 Parent(s): 1dbba77

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -109,14 +109,16 @@ For evaluation results, please refer to Section 3 and safety evaluations in Sect
109
  ## Example Code
110
 
111
  You can use the Code-Llama-2-7B-instruct-text2sql model to generate SQL queries from natural language questions, as demonstrated in the following code snippet:
 
 
 
112
 
113
  ```python
 
114
  from transformers import (
115
  AutoModelForCausalLM,
116
- AutoTokenizer,
117
- pipeline
118
  )
119
- import torch
120
 
121
  model_name = 'support-pvelocity/Code-Llama-2-7B-instruct-text2sql'
122
 
 
109
  ## Example Code
110
 
111
  You can use the Code-Llama-2-7B-instruct-text2sql model to generate SQL queries from natural language questions, as demonstrated in the following code snippet:
112
+ ```cmd
113
+ pip install -q accelerate transformers
114
+ ```
115
 
116
  ```python
117
+ import torch
118
  from transformers import (
119
  AutoModelForCausalLM,
120
+ AutoTokenizer
 
121
  )
 
122
 
123
  model_name = 'support-pvelocity/Code-Llama-2-7B-instruct-text2sql'
124