Oscar Wu
commited on
Commit
•
d89aa45
1
Parent(s):
97f1188
Updated README
Browse files
README.md
CHANGED
|
@@ -29,7 +29,9 @@ Loading the model checkpoint of this xMADified model requires less than 12 GiB o
|
|
| 29 |
|
| 30 |
**Package prerequisites**: Run the following commands to install the required packages.
|
| 31 |
```bash
|
| 32 |
-
pip install torch==2.4.0
|
|
|
|
|
|
|
| 33 |
```
|
| 34 |
|
| 35 |
**Sample Inference Code**
|
|
|
|
| 29 |
|
| 30 |
**Package prerequisites**: Run the following commands to install the required packages.
|
| 31 |
```bash
|
| 32 |
+
pip install torch==2.4.0 # Run following if you have CUDA version 11.8: pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cu118
|
| 33 |
+
pip install transformers accelerate optimum
|
| 34 |
+
pip install -vvv --no-build-isolation "git+https://github.com/PanQiWei/[email protected]"
|
| 35 |
```
|
| 36 |
|
| 37 |
**Sample Inference Code**
|