annabeth97c commited on
Commit
2d6ee8c
·
verified ·
1 Parent(s): 27deb4c

Update requirements to be compatible with zeroGPU

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -90,7 +90,7 @@ nvidia-cufft-cu12==11.0.2.54
90
  nvidia-curand-cu12==10.3.2.106
91
  nvidia-cusolver-cu12==11.4.5.107
92
  nvidia-cusparse-cu12==12.1.0.106
93
- nvidia-nccl-cu12==2.20.5
94
  nvidia-nvjitlink-cu12==12.5.82
95
  nvidia-nvtx-cu12==12.1.105
96
  openai==1.82.0
@@ -146,14 +146,14 @@ tensorboard-data-server==0.7.2
146
  termcolor==2.4.0
147
  threadpoolctl==3.5.0
148
  tokenizers==0.19.1
149
- torch==2.3.1
150
  torch-stoi==0.2.1
151
- torchaudio==2.3.1
152
  torchviz==0.0.2
153
  tqdm==4.66.4
154
  traitlets==5.14.3
155
  transformers==4.40.1
156
- triton==2.3.1
157
  typing_extensions==4.12.2
158
  tzdata==2024.1
159
  urllib3==2.2.2
 
90
  nvidia-curand-cu12==10.3.2.106
91
  nvidia-cusolver-cu12==11.4.5.107
92
  nvidia-cusparse-cu12==12.1.0.106
93
+ nvidia-nccl-cu12==2.19.3
94
  nvidia-nvjitlink-cu12==12.5.82
95
  nvidia-nvtx-cu12==12.1.105
96
  openai==1.82.0
 
146
  termcolor==2.4.0
147
  threadpoolctl==3.5.0
148
  tokenizers==0.19.1
149
+ torch==2.2.0
150
  torch-stoi==0.2.1
151
+ torchaudio==2.2.0
152
  torchviz==0.0.2
153
  tqdm==4.66.4
154
  traitlets==5.14.3
155
  transformers==4.40.1
156
+ triton==2.2.0
157
  typing_extensions==4.12.2
158
  tzdata==2024.1
159
  urllib3==2.2.2