Explorar o código

added requirements-cuda

Jhjoon05 %!s(int64=2) %!d(string=hai) anos
pai
achega
4d25b99548
Modificáronse 1 ficheiros con 7 adicións e 0 borrados
  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