Spaces:
Sleeping
Sleeping
File size: 2,159 Bytes
c5ed230 d94ccbe c5ed230 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
INSTALL_DIR=/usr/local/vits-simple-api
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
PLAIN='\033[0m'
mkdir -p $INSTALL_DIR
cd $INSTALL_DIR
if [ ! -f config.py ]; then
echo -e "${YELLOW}download config.py\n${PLAIN}"
wget -O $INSTALL_DIR/config.py https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/config.py
fi
if [ ! -f gunicorn_config.py ]; then
echo -e "${YELLOW}download config.py\n${PLAIN}"
wget -O $INSTALL_DIR/gunicorn_config.py https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/gunicorn_config.py
fi
while true; do
echo -e "${GREEN}Which version of docker-compose.yaml do you want to download?"
echo -e "1. docker-compose.yaml (CPU version)"
echo -e "2. docker-compose-gpu.yaml (GPU version)"
read -p "Enter your choice (1 or 2): " choice
case $choice in
1)
echo -e "${YELLOW}Downloading docker-compose.yaml (CPU version)\n${PLAIN}"
wget -O $INSTALL_DIR/docker-compose.yaml https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/docker-compose.yaml
break
;;
2)
echo -e "${YELLOW}Downloading docker-compose-gpu.yaml (GPU version)\n${PLAIN}"
wget -O $INSTALL_DIR/docker-compose.yaml https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/docker-compose-gpu.yaml
break
;;
*)
echo -e "${RED}Invalid choice. Please enter 1 or 2.${PLAIN}"
;;
esac
done
echo -e "${YELLOW}Pulling the image might take a while, so why not grab a cup of java first?\n${PLAIN}"
docker compose pull
docker compose up -d
echo -e "\nThe upgrade or installation has been completed."
echo -e "The configuration file directory is $(realpath $INSTALL_DIR)"
echo -e "${YELLOW}If the vits model is not imported, it cannot be used. Import the model in the configuration file directory.${PLAIN}"
echo -e "After modifying the configuration file, restart the docker container for the modification to take effect."
echo -e "${YELLOW}If you have any questions, please put them in the issues.${PLAIN}"
echo -e "https://github.com/Artrajz/vits-simple-api" |