浏览代码

switch to torch 1.12

justheuristic 3 年之前
父节点
当前提交
56cbd088ed
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      requirements.txt

+ 1 - 1
requirements.txt

@@ -1,4 +1,4 @@
-torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
+torch==1.12.0
 accelerate==0.10.0
 huggingface-hub==0.7.0
 bitsandbytes-cuda113==0.26.0