Skip to content

Commit a60b40f

Browse files
authored
Dev (#376)
* MAC-OS-SUPPORT * MAC-OS-SUPPORT-UPD * Update app.py * Update README.md * Update README_RU.md * Update README_ZH.md * Update README.md * Update README_RU.md * Update README_ZH.md
1 parent 929347a commit a60b40f

11 files changed

+920
-222
lines changed

Install.sh

Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,18 +4,26 @@ CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &> /dev/null && pwd)"
44

55
echo "Select installation type:"
66
echo "1. GPU"
7-
echo "2. CPU"
7+
echo "2. CPU OR MPS"
88
read -n 1 -p "Enter number (1 or 2): " choice
99
echo ""
1010

1111
if [ "$choice" = "2" ]; then
1212
INSTALL_TYPE="CPU"
1313
export BUILD_CUDA_EXT=0
1414
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
1522
else
1623
INSTALL_TYPE="GPU"
1724
export BUILD_CUDA_EXT=1
1825
export INSTALL_KERNELS=1
26+
MPS_MODE=false
1927
fi
2028

2129
clear
@@ -44,10 +52,17 @@ ERROR_LOG="$CURRENT_DIR/TechnicalFiles/logs/installation_errors.log"
4452
touch "$ERROR_LOG"
4553

4654
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
5166
else
5267
pip install --no-deps -r "$CURRENT_DIR/RequirementsFiles/requirements.txt" 2>> "$ERROR_LOG"
5368
pip install --no-deps -r "$CURRENT_DIR/RequirementsFiles/requirements-cuda.txt" 2>> "$ERROR_LOG"

0 commit comments

Comments
 (0)