Update README.md
Browse files
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:
|