ctranslate2-4you commited on
Commit
44e4123
·
verified ·
1 Parent(s): 9f6505d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -2
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 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
 
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