Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ https://huggingface.co/allenai/Molmo-7B-D-0924
|
|
16 |
|
17 |
NOTE:
|
18 |
|
19 |
-
The sample script below requires that you install the following libraries into your virtual environment and
|
20 |
|
21 |
- ```nvidia-cublas-cu12==12.1.3.1```
|
22 |
- ```nvidia-cuda-runtime-cu12==12.1.105```
|
@@ -25,7 +25,7 @@ The sample script below requires that you install the following libraries into y
|
|
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 this command:
|
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 |
|
|
|
16 |
|
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```
|
|
|
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 this command:
|
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 |
|