Explorar o código

Update torch for ARM Macs to 2.3.1

viking1304 hai 1 ano
pai
achega
a772fd9804
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      webui-macos-env.sh

+ 1 - 1
webui-macos-env.sh

@@ -16,7 +16,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1
 if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
 if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
     export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
     export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
 else
 else
-    export TORCH_COMMAND="pip install torch==2.3.0 torchvision==0.18.0"
+    export TORCH_COMMAND="pip install torch==2.3.1 torchvision==0.18.1"
 fi
 fi
 
 
 ####################################################################
 ####################################################################