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