@@ -4,18 +4,26 @@ CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &> /dev/null && pwd)"
4
4
5
5
echo " Select installation type:"
6
6
echo " 1. GPU"
7
- echo " 2. CPU"
7
+ echo " 2. CPU OR MPS "
8
8
read -n 1 -p " Enter number (1 or 2): " choice
9
9
echo " "
10
10
11
11
if [ " $choice " = " 2" ]; then
12
12
INSTALL_TYPE=" CPU"
13
13
export BUILD_CUDA_EXT=0
14
14
export INSTALL_KERNELS=0
15
+
16
+ if system_profiler SPDisplaysDataType | grep -q " Metal" ; then
17
+ echo " MPS is detected. Installing MPS-specific requirements."
18
+ MPS_MODE=true
19
+ else
20
+ MPS_MODE=false
21
+ fi
15
22
else
16
23
INSTALL_TYPE=" GPU"
17
24
export BUILD_CUDA_EXT=1
18
25
export INSTALL_KERNELS=1
26
+ MPS_MODE=false
19
27
fi
20
28
21
29
clear
@@ -44,10 +52,17 @@ ERROR_LOG="$CURRENT_DIR/TechnicalFiles/logs/installation_errors.log"
44
52
touch " $ERROR_LOG "
45
53
46
54
if [ " $INSTALL_TYPE " = " CPU" ]; then
47
- pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-СPU.txt" 2>> " $ERROR_LOG "
48
- pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-cuda-CPU.txt" 2>> " $ERROR_LOG "
49
- pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-llama-cpp-CPU.txt" 2>> " $ERROR_LOG "
50
- pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-stable-diffusion-cpp-CPU.txt" 2>> " $ERROR_LOG "
55
+ if [ " $MPS_MODE " = true ]; then
56
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-СPU.txt" 2>> " $ERROR_LOG "
57
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-cuda-CPU.txt" 2>> " $ERROR_LOG "
58
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-llama-cpp-MPS.txt" 2>> " $ERROR_LOG "
59
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-stable-diffusion-cpp-MPS.txt" 2>> " $ERROR_LOG "
60
+ else
61
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-СPU.txt" 2>> " $ERROR_LOG "
62
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-cuda-CPU.txt" 2>> " $ERROR_LOG "
63
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-llama-cpp-CPU.txt" 2>> " $ERROR_LOG "
64
+ pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-stable-diffusion-cpp-CPU.txt" 2>> " $ERROR_LOG "
65
+ fi
51
66
else
52
67
pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements.txt" 2>> " $ERROR_LOG "
53
68
pip install --no-deps -r " $CURRENT_DIR /RequirementsFiles/requirements-cuda.txt" 2>> " $ERROR_LOG "
0 commit comments