support-pvelocity commited on
Commit
23f8d2d
·
1 Parent(s): 3383f0d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -52,6 +52,10 @@ This model is governed by a custom commercial license from Code Llama. For detai
52
  ## Example Code
53
 
54
  You can use the Code-Llama-2-13B-instruct-text2sql-GPTQ model to generate SQL queries from natural language questions, as demonstrated in the following code snippet:
 
 
 
 
55
  ```python
56
  from transformers import AutoTokenizer
57
  from auto_gptq import AutoGPTQForCausalLM
 
52
  ## Example Code
53
 
54
  You can use the Code-Llama-2-13B-instruct-text2sql-GPTQ model to generate SQL queries from natural language questions, as demonstrated in the following code snippet:
55
+ ```cmd
56
+ pip install -q transformers==4.35.0 torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 optimum==1.13.2 auto-gptq==0.4.2
57
+ ```
58
+
59
  ```python
60
  from transformers import AutoTokenizer
61
  from auto_gptq import AutoGPTQForCausalLM