Spaces:
Sleeping
Sleeping
File size: 11,669 Bytes
c5ed230 35c29ec c5ed230 35c29ec c5ed230 35c29ec c5ed230 d94ccbe 35c29ec d94ccbe 35c29ec c5ed230 35c29ec c5ed230 35c29ec c5ed230 35c29ec |
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 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 |
INSTALL_DIR=/usr/local/vits-simple-api
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
PLAIN='\033[0m'
declare -A EN_MESSAGES
declare -A ZH_MESSAGES
EN_MESSAGES=(
["ATTEMPT_DOWNLOAD"]="Attempting to download"
["FROM"]="from"
["DOWNLOAD_FAIL"]="Failed to download"
["FROM_ALL_URLS"]="from all provided URLs."
["DOWNLOADING"]="Downloading..."
["VERIFYING"]="Verifying..."
["UNZIPPING"]="Unzipping..."
["CHOOSE_VERSION"]="Which version of docker-compose.yaml do you want to download?"
["DOCKER_CPU"]="docker-compose.yaml (CPU version)"
["DOCKER_GPU"]="docker-compose-gpu.yaml (GPU version)"
["ENTER_CHOICE"]="Enter your choice (1 or 2): "
["INVALID_CHOICE"]="Invalid choice. Please enter 1 or 2."
["DOWNLOAD_CONFIG"]="Downloading configuration file shortly..."
["PULL_IMAGE"]="Do you want to start pulling the image? Enter 1 for yes or 2 for no"
["DOWNLOAD_DICT"]="Do you want to download the pyopenjtalk dictionary file? Enter 1 for yes or 2 for no"
["MUST_DOWNLOAD_JP"]="Japanese model must be downloaded."
["DOWNLOAD_VITS_CHINESE"]="Do you want to download the bert model for vits_chinese? Enter 1 for yes, 2 for no."
["MUST_DOWNLOAD_VITS_CHINESE"]="Using vits_chinese requires downloading these models, which will take up about 410MB."
["DOWNLOAD_BERT_VITS2"]="Do you want to download chinese-roberta-wwm-ext-large? Enter 1 for yes or 2 for no"
["MUST_DOWNLOAD_BERT_VITS2"]="To use Bert-VITS2, you must download these models, which will take up about 3.64GB."
["DOWNLOADED"]="File is downloaded correctly."
["CORRUPTED"]="File is corrupted or incomplete."
["INSTALL_COMPLETE"]="The upgrade or installation has been completed."
["CONFIG_DIR"]="The configuration file directory is"
["IMPORT_NOTICE"]="If the vits model is not imported, it cannot be used. Import the model in the configuration file directory."
["RESTART_NOTICE"]="After modifying the configuration file, restart the docker container for the modification to take effect."
["ISSUE_NOTICE"]="If you have any questions, please put them in the issues."
["GITHUB_LINK"]="https://github.com/Artrajz/vits-simple-api"
)
ZH_MESSAGES=(
["ATTEMPT_DOWNLOAD"]="正在尝试下载"
["FROM"]="从"
["DOWNLOAD_FAIL"]="都下载失败"
["FROM_ALL_URLS"]="从所有提供的URLs"
["DOWNLOADING"]="正在下载..."
["VERIFYING"]="正在校验"
["UNZIPPING"]="正在解压..."
["CHOOSE_VERSION"]="你想下载哪个版本的docker-compose.yaml?"
["DOCKER_CPU"]="docker-compose.yaml (CPU版本)"
["DOCKER_GPU"]="docker-compose-gpu.yaml (GPU版本)"
["ENTER_CHOICE"]="请输入您的选择 (1 或 2): "
["INVALID_CHOICE"]="无效选择。 请重新输入 1 或 2。"
["DOWNLOAD_CONFIG"]="即将下载配置文件..."
["PULL_IMAGE"]="是否要开始拉取镜像?输入1表示是,2表示否。"
["DOWNLOAD_DICT"]="是否要下载pyopenjtalk的词典文件?输入1表示是,2表示否。"
["MUST_DOWNLOAD_JP"]="使用日语模型必须下载该词典文件,将占用大约102MB。"
["DOWNLOAD_VITS_CHINESE"]="是否要下载vits_chinese的bert模型?输入1表示是,2表示否。"
["MUST_DOWNLOAD_VITS_CHINESE"]="使用vits_chinese必须下载这些模型,将占用大约410MB。"
["DOWNLOAD_BERT_VITS2"]="是否要下载chinese-roberta-wwm-ext-large?输入1表示是,2表示否。"
["MUST_DOWNLOAD_BERT_VITS2"]="使用Bert-VITS2必须下载这些模型,将占用大约3.64GB。"
["DOWNLOADED"]="文件已正确下载。"
["CORRUPTED"]="文件已损坏或不完整。"
["INSTALL_COMPLETE"]="更新或安装已完成。"
["CONFIG_DIR"]="配置文件目录是"
["IMPORT_NOTICE"]="如果vits模型没有被导入,它是无法使用的。请在配置文件目录中导入模型。"
["RESTART_NOTICE"]="修改配置文件后,请重启docker容器以使修改生效。"
["ISSUE_NOTICE"]="如果你有任何问题,请在issues中提出,或者加入q群提问。"
["GITHUB_LINK"]="https://github.com/Artrajz/vits-simple-api"
)
echo -e "${PLAIN}${GREEN}Choose a language/选择语言: ${PLAIN}"
echo "1. English"
echo "2. 中文"
read -p "Enter your choice (1 or 2): " choice_language
declare -A MESSAGES
if [ "$choice_language" -eq 1 ]; then
for key in "${!EN_MESSAGES[@]}"; do
MESSAGES["$key"]="${EN_MESSAGES[$key]}"
done
else
for key in "${!ZH_MESSAGES[@]}"; do
MESSAGES["$key"]="${ZH_MESSAGES[$key]}"
done
fi
mkdir -p $INSTALL_DIR
cd $INSTALL_DIR
download_with_fallback() {
local filename=$1
shift # Shift arguments to the left to handle URLs
local success=0
local url
for url in "$@"; do
echo -e "${YELLOW}${MESSAGES["ATTEMPT_DOWNLOAD"]} $filename ${MESSAGES["FROM"]} $url\n${PLAIN}"
if wget -O "$INSTALL_DIR/$filename" "$url"; then
success=1
break
fi
done
if [ "$success" -ne 1 ]; then
echo -e "${RED} $filename ${MESSAGES["FROM_ALL_URLS"]} ${MESSAGES["DOWNLOAD_FAIL"]}${PLAIN}"
exit 1
fi
}
version_gt() {
test "$(echo "$@" | tr " " "\n" | sort -V | head -n 1)" != "$1"
}
while true; do
echo -e "${GREEN}${MESSAGES["CHOOSE_VERSION"]}${PLAIN}"
echo -e "1. ${MESSAGES["DOCKER_CPU"]}"
echo -e "2. ${MESSAGES["DOCKER_GPU"]}"
read -p "${MESSAGES["ENTER_CHOICE"]}" choice_gpu
case $choice_gpu in
1)
echo -e "${MESSAGES["DOWNLOADING"]}"
download_with_fallback docker-compose.yaml \
"https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/docker-compose.yaml" \
"https://ghproxy.com/https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/docker-compose.yaml"
break
;;
2)
echo -e "${MESSAGES["DOWNLOADING"]}"
download_with_fallback docker-compose.yaml \
"https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/docker-compose-gpu.yaml" \
"https://ghproxy.com/https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/docker-compose-gpu.yaml"
break
;;
*)
echo -e "${RED}${MESSAGES["INVALID_CHOICE"]}${PLAIN}"
;;
esac
done
if [ "$choice_gpu" -eq 2 ]; then
DOCKER_VERSION=$(docker version --format '{{.Server.Version}}')
MIN_DOCKER_VERSION="19.03"
if version_gt $MIN_DOCKER_VERSION $DOCKER_VERSION; then
echo -e "${RED}Your Docker version ($DOCKER_VERSION) does not support GPU. You need at least version $MIN_DOCKER_VERSION.${PLAIN}"
exit 1
fi
fi
if ! command -v docker-compose &>/dev/null; then
echo -e "${RED}docker-compose could not be found.${PLAIN}"
exit 1
fi
echo -e "${GREEN}${MESSAGES["PULL_IMAGE"]}${PLAIN}"
read -p "${MESSAGES["ENTER_CHOICE"]}" choice_pull
if [ "$choice_pull" -eq 1 ]; then
docker compose pull
docker compose up -d
fi
echo -e "${YELLOW}${MESSAGES["DOWNLOAD_CONFIG"]}${PLAIN}"
if [ ! -f config.py ]; then
download_with_fallback config.py \
"https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/config.py" \
"https://ghproxy.com/https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/config.py"
fi
if [ ! -f gunicorn_config.py ]; then
download_with_fallback gunicorn_config.py \
"https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/gunicorn_config.py" \
"https://ghproxy.com/https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/gunicorn_config.py"
fi
download_with_fallback config.example.py \
"https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/config.py" \
"https://ghproxy.com/https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/config.py"
download_with_fallback gunicorn_config.example.py \
"https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/gunicorn_config.py" \
"https://ghproxy.com/https://raw.githubusercontent.com/Artrajz/vits-simple-api/main/gunicorn_config.py"
echo -e "${GREEN}${MESSAGES["DOWNLOAD_DICT"]}${PLAIN}"
echo -e "${GREEN}${MESSAGES["MUST_DOWNLOAD_JP"]}${PLAIN}"
read -p "${MESSAGES["ENTER_CHOICE"]}" choice_download_pyopenjtalk
if [ "$choice_download_pyopenjtalk" -eq 1 ]; then
mkdir -p pyopenjtalk
echo -e "${MESSAGES["DOWNLOADING"]}"
download_with_fallback open_jtalk_dic_utf_8-1.11.tar.gz \
"https://github.com/r9y9/open_jtalk/releases/download/v1.11.1/open_jtalk_dic_utf_8-1.11.tar.gz" \
"https://ghproxy.com/https://github.com/r9y9/open_jtalk/releases/download/v1.11.1/open_jtalk_dic_utf_8-1.11.tar.gz"
echo -e "${MESSAGES["UNZIPPING"]}"
tar -xzvf open_jtalk_dic_utf_8-1.11.tar.gz -C pyopenjtalk/
rm open_jtalk_dic_utf_8-1.11.tar.gz
fi
echo -e "${GREEN}${MESSAGES["DOWNLOAD_VITS_CHINESE"]}${PLAIN}"
echo -e "${GREEN}${MESSAGES["MUST_DOWNLOAD_VITS_CHINESE"]}${PLAIN}"
read -p "${MESSAGES["ENTER_CHOICE"]}" choice_download_vits_chinese
if [ "$choice_download_vits_chinese" -eq 1 ]; then
mkdir -p vits/bert
EXPECTED_MD5="dea78034433141adc8002404aa1b3184"
FILE_PATH="vits/bert/prosody_model.pt"
echo -e "${MESSAGES["VERIFYING"]}$FILE_PATH"
ACTUAL_MD5=$(md5sum $FILE_PATH | awk '{print $1}')
if [ "$EXPECTED_MD5" == "$ACTUAL_MD5" ]; then
echo "${MESSAGES["DOWNLOADED"]}"
else
echo "${MESSAGES["CORRUPTED"]}"
download_with_fallback vits/bert/prosody_model.pt \
"https://huggingface.co/spaces/maxmax20160403/vits_chinese/resolve/main/bert/prosody_model.pt"
fi
fi
echo -e "${GREEN}${MESSAGES["DOWNLOAD_BERT_VITS2"]}${PLAIN}"
echo -e "${GREEN}${MESSAGES["MUST_DOWNLOAD_BERT_VITS2"]}${PLAIN}"
read -p "${MESSAGES["ENTER_CHOICE"]}" choice_download_bert_vits2
if [ "$choice_download_bert_vits2" -eq 1 ]; then
mkdir -p bert_vits2/bert/chinese-roberta-wwm-ext-large
EXPECTED_MD5="78ef42421495cb23372bef9d069a75f3"
FILE_PATH="bert_vits2/bert/chinese-roberta-wwm-ext-large/flax_model.msgpack"
echo -e "${MESSAGES["VERIFYING"]}$FILE_PATH"
ACTUAL_MD5=$(md5sum $FILE_PATH | awk '{print $1}')
if [ "$EXPECTED_MD5" == "$ACTUAL_MD5" ]; then
echo "${MESSAGES["DOWNLOADED"]}"
else
echo "${MESSAGES["CORRUPTED"]}"
download_with_fallback bert_vits2/bert/chinese-roberta-wwm-ext-large/flax_model.msgpack \
"https://huggingface.co/hfl/chinese-roberta-wwm-ext-large/resolve/main/flax_model.msgpack"
fi
EXPECTED_MD5="15d7435868fef1bd4222ff7820149a2a"
FILE_PATH="bert_vits2/bert/chinese-roberta-wwm-ext-large/pytorch_model.bin"
echo -e "${MESSAGES["VERIFYING"]}$FILE_PATH"
ACTUAL_MD5=$(md5sum $FILE_PATH | awk '{print $1}')
if [ "$EXPECTED_MD5" == "$ACTUAL_MD5" ]; then
echo "${MESSAGES["DOWNLOADED"]}"
else
echo ${MESSAGES["CORRUPTED"]}
download_with_fallback bert_vits2/bert/chinese-roberta-wwm-ext-large/pytorch_model.bin \
"https://huggingface.co/hfl/chinese-roberta-wwm-ext-large/resolve/main/pytorch_model.bin"
fi
EXPECTED_MD5="d15991416bd4a86fa127c70d3c0f4779"
FILE_PATH="bert_vits2/bert/chinese-roberta-wwm-ext-large/tf_model.h5"
echo -e "${MESSAGES["VERIFYING"]}$FILE_PATH"
ACTUAL_MD5=$(md5sum $FILE_PATH | awk '{print $1}')
if [ "$EXPECTED_MD5" == "$ACTUAL_MD5" ]; then
echo "${MESSAGES["DOWNLOADED"]}"
else
echo "${MESSAGES["CORRUPTED"]}"
download_with_fallback bert_vits2/bert/chinese-roberta-wwm-ext-large/tf_model.h5 \
"https://huggingface.co/hfl/chinese-roberta-wwm-ext-large/resolve/main/tf_model.h5"
fi
fi
if [ "$choice_gpu" -eq 2 ]; then
if ! docker run --gpus all artrajz/vits-simple-api:latest-gpu nvidia-smi &>/dev/null; then
echo -e "${RED}Your Docker does not seem to support GPU or NVIDIA Docker is not installed properly.${PLAIN}"
exit 1
fi
fi
echo -e "\n${MESSAGES["INSTALL_COMPLETE"]}"
echo -e "${MESSAGES["CONFIG_DIR"]} $(realpath $INSTALL_DIR)"
echo -e "${YELLOW}${MESSAGES["IMPORT_NOTICE"]}${PLAIN}"
echo -e "${YELLOW}${MESSAGES["RESTART_NOTICE"]}${PLAIN}"
echo -e "${MESSAGES["ISSUE_NOTICE"]}"
echo -e "${MESSAGES["GITHUB_LINK"]}"
|