Parcourir la source

switch to torch 1.12

justheuristic il y a 3 ans
Parent
commit
56cbd088ed
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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