Browse Source

added requirements-cuda

Jhjoon05 2 năm trước cách đây
mục cha
commit
4d25b99548
1 tập tin đã thay đổi với 7 bổ sung0 xóa
  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