瀏覽代碼

added requirements-cuda

Jhjoon05 2 年之前
父節點
當前提交
4d25b99548
共有 1 個文件被更改,包括 7 次插入0 次删除
  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