|
@@ -11,7 +11,12 @@ fi
|
|
|
|
|
|
export install_dir="$HOME"
|
|
export install_dir="$HOME"
|
|
export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
|
|
export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
|
|
-export TORCH_COMMAND="pip install torch==2.1.0 torchvision==0.16.0"
|
|
|
|
export PYTORCH_ENABLE_MPS_FALLBACK=1
|
|
export PYTORCH_ENABLE_MPS_FALLBACK=1
|
|
|
|
|
|
|
|
+if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
|
|
|
|
+ export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
|
|
|
|
+else
|
|
|
|
+ export TORCH_COMMAND="pip install torch==2.3.0 torchvision==0.18.0"
|
|
|
|
+fi
|
|
|
|
+
|
|
####################################################################
|
|
####################################################################
|