jerryzh168 commited on
Commit
e60e06a
·
verified ·
1 Parent(s): 626861a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -27,6 +27,8 @@ First need to install the required packages:
27
  ```
28
  pip install git+https://github.com/huggingface/transformers@main
29
  pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126
 
 
30
  ```
31
 
32
  We used following code to get the quantized model:
 
27
  ```
28
  pip install git+https://github.com/huggingface/transformers@main
29
  pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126
30
+ pip install torch
31
+ pip install accelerate
32
  ```
33
 
34
  We used following code to get the quantized model: