jerryzh168 commited on
Commit
ea29a3b
·
verified ·
1 Parent(s): e3d011c

Update README.md

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