Explorar o código

switch to torch 1.12

justheuristic %!s(int64=3) %!d(string=hai) anos
pai
achega
56cbd088ed
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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