Update README.md
Browse files
README.md
CHANGED
@@ -17,17 +17,22 @@ https://huggingface.co/allenai/Molmo-7B-D-0924
|
|
17 |
NOTE:
|
18 |
|
19 |
The sample script below requires that you install the following libraries into your virtual environment, and an Nvidia GPU is required. You can rely on a system-wide installation of CUDA, in which case just remove the ```set_cuda_paths``` function.
|
20 |
-
|
21 |
- ```nvidia-cublas-cu12==12.1.3.1```
|
22 |
- ```nvidia-cuda-runtime-cu12==12.1.105```
|
23 |
- ```nvidia-cuda-nvrtc-cu12==12.1.105```
|
24 |
- ```nvidia-cufft-cu12==11.0.2.54```
|
25 |
|
26 |
-
Also, I've only tested it on ```torch==2.2.2``` although I plan on testing with higher versions. For Windows users, install torch with
|
27 |
|
28 |
```
|
29 |
pip install https://download.pytorch.org/whl/cu121/torch-2.2.2%2Bcu121-cp311-cp311-win_amd64.whl#sha256=efbcfdd4399197d06b32f7c0e1711c615188cdd65427b933648c7478fb880b3f
|
30 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
|
32 |
Example script (process single image):
|
33 |
|
|
|
17 |
NOTE:
|
18 |
|
19 |
The sample script below requires that you install the following libraries into your virtual environment, and an Nvidia GPU is required. You can rely on a system-wide installation of CUDA, in which case just remove the ```set_cuda_paths``` function.
|
|
|
20 |
- ```nvidia-cublas-cu12==12.1.3.1```
|
21 |
- ```nvidia-cuda-runtime-cu12==12.1.105```
|
22 |
- ```nvidia-cuda-nvrtc-cu12==12.1.105```
|
23 |
- ```nvidia-cufft-cu12==11.0.2.54```
|
24 |
|
25 |
+
Also, I've only tested it on ```torch==2.2.2``` although I plan on testing with higher versions. For Windows users, install torch with these commands:
|
26 |
|
27 |
```
|
28 |
pip install https://download.pytorch.org/whl/cu121/torch-2.2.2%2Bcu121-cp311-cp311-win_amd64.whl#sha256=efbcfdd4399197d06b32f7c0e1711c615188cdd65427b933648c7478fb880b3f
|
29 |
```
|
30 |
+
```
|
31 |
+
pip install https://download.pytorch.org/whl/cu121/torchvision-0.17.2%2Bcu121-cp311-cp311-win_amd64.whl#sha256=10ad542aab6b47dbe73c441381986d50a7ed5021cbe01d593a14477ec1f067a0
|
32 |
+
```
|
33 |
+
```
|
34 |
+
pip install https://download.pytorch.org/whl/cu121/torchaudio-2.2.2%2Bcu121-cp311-cp311-win_amd64.whl#sha256=c7dee68cd3d2b889bab71d4a0c345bdc3ea2fe79a62b921a6b49292c605b6071
|
35 |
+
```
|
36 |
|
37 |
Example script (process single image):
|
38 |
|