sunxiaokang commited on
Commit
a1862b1
·
verified ·
1 Parent(s): fadca31

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -3,7 +3,7 @@ pytorch==2.1.2
3
  torchvision==0.16.2
4
  torchaudio==2.1.2
5
  termcolor==2.4.0
6
- pytorch-cuda=12.1
7
  class_registry==2.1.2
8
  datasets==3.1.0
9
  griffe==0.48.0
 
3
  torchvision==0.16.2
4
  torchaudio==2.1.2
5
  termcolor==2.4.0
6
+ pytorch-cuda==12.1
7
  class_registry==2.1.2
8
  datasets==3.1.0
9
  griffe==0.48.0