瀏覽代碼

Upgrade torch requirement to 2.1 RC

Matthias Reso 1 年之前
父節點
當前提交
bf152a7dcb
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      requirements.txt

+ 1 - 1
requirements.txt

@@ -1,4 +1,4 @@
--f https://download.pytorch.org/whl/torch_stable.html torch==2.0.1+cu118
+--index-url https://download.pytorch.org/whl/test/cu118 torch 
 accelerate
 appdirs
 loralib