Rishi Desai commited on
Commit
9f37da3
·
1 Parent(s): 3d720fd

req with torch 2.4.0

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -17
requirements.txt CHANGED
@@ -76,19 +76,19 @@ multidict==6.4.3
76
  networkx==3.4.2
77
  numba==0.61.2
78
  numpy==2.2.4
79
- nvidia-cublas-cu12==12.4.5.8
80
- nvidia-cuda-cupti-cu12==12.4.127
81
- nvidia-cuda-nvrtc-cu12==12.4.127
82
- nvidia-cuda-runtime-cu12==12.4.127
83
  nvidia-cudnn-cu12==9.1.0.70
84
- nvidia-cufft-cu12==11.2.1.3
85
- nvidia-curand-cu12==10.3.5.147
86
- nvidia-cusolver-cu12==11.6.1.9
87
- nvidia-cusparse-cu12==12.3.1.170
88
  nvidia-cusparselt-cu12==0.6.2
89
- nvidia-nccl-cu12==2.21.5
90
- nvidia-nvjitlink-cu12==12.4.127
91
- nvidia-nvtx-cu12==12.4.127
92
  onnx==1.17.0
93
  onnxruntime==1.21.0
94
  onnxruntime-gpu==1.21.0
@@ -104,7 +104,7 @@ prettytable==3.16.0
104
  prompt_toolkit==3.0.51
105
  propcache==0.3.1
106
  protobuf==6.30.2
107
- psutil==7.0.0
108
  pycparser==2.22
109
  pydantic==2.11.3
110
  pydantic_core==2.33.1
@@ -140,6 +140,7 @@ six==1.17.0
140
  smmap==5.0.2
141
  sniffio==1.3.1
142
  soundfile==0.13.1
 
143
  spandrel==0.4.1
144
  starlette==0.46.2
145
  stringzilla==3.12.4
@@ -151,14 +152,14 @@ timm==1.0.15
151
  tokenizers==0.21.1
152
  toml==0.10.2
153
  tomlkit==0.13.2
154
- torch==2.6.0
155
- torchaudio==2.6.0
156
  torchsde==0.2.6
157
- torchvision==0.21.0
158
  tqdm==4.67.1
159
  trampoline==0.1.2
160
  transformers==4.51.3
161
- triton==3.2.0
162
  typer==0.15.2
163
  types-python-dateutil==2.9.0.20241206
164
  typing-inspection==0.4.0
@@ -173,4 +174,4 @@ wcwidth==0.2.13
173
  websocket-client==1.8.0
174
  websockets==15.0.1
175
  wrapt==1.17.2
176
- yarl==1.20.0
 
76
  networkx==3.4.2
77
  numba==0.61.2
78
  numpy==2.2.4
79
+ nvidia-cublas-cu12==12.4.2.65
80
+ nvidia-cuda-cupti-cu12==12.4.99
81
+ nvidia-cuda-nvrtc-cu12==12.4.99
82
+ nvidia-cuda-runtime-cu12==12.4.99
83
  nvidia-cudnn-cu12==9.1.0.70
84
+ nvidia-cufft-cu12==11.2.0.44
85
+ nvidia-curand-cu12==10.3.5.119
86
+ nvidia-cusolver-cu12==11.6.0.99
87
+ nvidia-cusparse-cu12==12.3.0.142
88
  nvidia-cusparselt-cu12==0.6.2
89
+ nvidia-nccl-cu12==2.20.5
90
+ nvidia-nvjitlink-cu12==12.4.99
91
+ nvidia-nvtx-cu12==12.4.99
92
  onnx==1.17.0
93
  onnxruntime==1.21.0
94
  onnxruntime-gpu==1.21.0
 
104
  prompt_toolkit==3.0.51
105
  propcache==0.3.1
106
  protobuf==6.30.2
107
+ psutil==5.9.8
108
  pycparser==2.22
109
  pydantic==2.11.3
110
  pydantic_core==2.33.1
 
140
  smmap==5.0.2
141
  sniffio==1.3.1
142
  soundfile==0.13.1
143
+ spaces==0.35.0
144
  spandrel==0.4.1
145
  starlette==0.46.2
146
  stringzilla==3.12.4
 
152
  tokenizers==0.21.1
153
  toml==0.10.2
154
  tomlkit==0.13.2
155
+ torch==2.4.0+cu124
156
+ torchaudio==2.4.0+cu124
157
  torchsde==0.2.6
158
+ torchvision==0.19.0+cu124
159
  tqdm==4.67.1
160
  trampoline==0.1.2
161
  transformers==4.51.3
162
+ triton==3.0.0
163
  typer==0.15.2
164
  types-python-dateutil==2.9.0.20241206
165
  typing-inspection==0.4.0
 
174
  websocket-client==1.8.0
175
  websockets==15.0.1
176
  wrapt==1.17.2
177
+ yarl==1.20.0