Browse Source

added requirements-cuda

Jhjoon05 2 years ago
parent
commit
4d25b99548
1 changed files with 7 additions and 0 deletions
  1. 7 0
      requirements-cuda.txt

+ 7 - 0
requirements-cuda.txt

@@ -0,0 +1,7 @@
+numpy
+--extra-index-url https://download.pytorch.org/whl/cu117
+torch
+tqdm
+more-itertools
+transformers>=4.19.0
+ffmpeg-python==0.2.0