GPT-SoVITS/GPT-SoVITS.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"provenance":[{"file_id":"https://github.com/RVC-Boss/GPT-SoVITS/blob/main/colab_webui.ipynb","timestamp":1711022039106}]},"kernelspec":{"name":"python3","display_name":"Python 3"}},"cells":[{"cell_type":"markdown","source":["\n","# 如果你需要挂载GoogleDrive，请点亮代码块右侧的Mounting_Google_Drive选项。\n","理论上代码块可以直接运行且只需运行一次，如果你遇到程序意外中断的情况请在Github的Issue区进行反馈。\n","\n","\n","---\n","\n","\n","代码执行执行时间预计为20分钟，请做好心理准备，但如果你已经在GoogleDrive中保存好了项目文件和模型，那么代码执行时间可能会缩短至10分钟。\n","\n","\n","---\n","\n","\n","如果你还没在GoogleDrive中保存项目文件但依然点亮了Mounting_Google_Drive选项，那么程序会自动在你的Drive中保存项目所需的所有文件和模型，以便于以后读取，模型和文件的总体积大概在5G左右，清注意管理存储空间。"],"metadata":{"id":"_o6a8GS2lWQM"}},{"cell_type":"code","source":["Mounting_Google_Drive = False  #@param {type:\"boolean\"}\n","\n","if Mounting_Google_Drive:\n","    # 挂载Google Drive\n","    from google.colab import drive\n","    import os\n","\n","    drive.mount('/content/drive')\n","\n","    # 检测 GPT-SoVITS 文件夹是否存在\n","    gpt_sovits_path = \"/content/drive/MyDrive/GPT-SoVITS\"\n","    if not os.path.exists(gpt_sovits_path):\n","        # 如果文件夹不存在，则克隆\n","        !pip install -q condacolab\n","        import condacolab\n","        condacolab.install_from_url(\"https://repo.anaconda.com/miniconda/Miniconda3-py39_23.11.0-2-Linux-x86_64.sh\")\n","        %cd -q /content/drive/MyDrive\n","        !git clone https://github.com/RVC-Boss/GPT-SoVITS\n","        gpt_sovits_exists = False\n","    else:\n","        # 如果文件夹存在，则更新\n","        %cd -q /content/drive/MyDrive/GPT-SoVITS\n","        !git pull origin main\n","        gpt_sovits_exists = True\n","\n","    %cd -q /content/drive/MyDrive/GPT-SoVITS\n","\n","    # 准备运行环境\n","    !conda create -n GPTSoVits python=3.9\n","    !conda init\n","    !conda activate GPTSoVits\n","    !conda install ipykernel\n","    !bash install.sh\n","\n","\n","    import shutil\n","\n","    pretrained_models_path = \"/content/drive/MyDrive/GPT-SoVITS/GPT_SoVITS/pretrained_models\"\n","    if not os.path.exists(pretrained_models_path + \"/GPT-SoVITS\"):\n","        %cd \"{pretrained_models_path}\"\n","        !git clone https://huggingface.co/lj1995/GPT-SoVITS\n","    else:\n","        %cd \"{pretrained_models_path}/GPT-SoVITS\"\n","        !git pull origin main\n","\n","    # 这行代码会导致模型占用更多空间，但可以避免一个棘手的问题\n","    if len(os.listdir(pretrained_models_path + \"/GPT-SoVITS\")) > 0:\n","        !cp -r \"/content/drive/MyDrive/GPT-SoVITS/GPT_SoVITS/pretrained_models/GPT-SoVITS/.\" \"/content/drive/MyDrive/GPT-SoVITS/GPT_SoVITS/pretrained_models/\"\n","\n","    # 创建ASR模型目录\n","    asr_models_path = \"/content/drive/MyDrive/GPT-SoVITS/tools/damo_asr/models\"\n","    if not os.path.exists(asr_models_path):\n","        os.makedirs(asr_models_path)\n","        %cd \"{asr_models_path}\"\n","        !git clone https://www.modelscope.cn/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch.git\n","        !git clone https://www.modelscope.cn/damo/speech_fsmn_vad_zh-cn-16k-common-pytorch.git\n","        !git clone https://www.modelscope.cn/damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch.git\n","    else:\n","        # 这部分代码不起作用，但不影响功能，暂时保留\n","        %cd \"{asr_models_path}/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch\"\n","        !git pull origin main\n","\n","        %cd \"{asr_models_path}/speech_fsmn_vad_zh-cn-16k-common-pytorch\"\n","        !git pull origin main\n","\n","        %cd \"{asr_models_path}/punc_ct-transformer_zh-cn-common-vocab272727-pytorch\"\n","        !git pull origin main\n","\n","    # 创建UV5模型目录\n","    uvr5_models_path = \"/content/drive/MyDrive/GPT-SoVITS/tools/uvr5\"\n","    if not os.path.exists(uvr5_models_path + \"/uvr5_weights\"):\n","        os.makedirs(uvr5_models_path)\n","        %cd \"{uvr5_models_path}\"\n","        !git clone https://huggingface.co/Delik/uvr5_weights\n","        !git config core.sparseCheckout true\n","    else:\n","        %cd \"{uvr5_models_path}/uvr5_weights\"\n","        !git pull origin main\n","\n","    # 修改配置文件中的值\n","    !sed -i '10s/False/True/' \"/content/drive/MyDrive/GPT-SoVITS/config.py\"\n","\n","    # 进入 GPT-SoVITS 目录并启动 WebUI\n","    %cd \"/content/drive/MyDrive/GPT-SoVITS/\"\n","    !/usr/local/bin/python webui.py\n","\n","else:\n","\n","    # 以下为初版colab notebook的代码，唯一的区别是原先有关运行环境的代码会导致runtime重启，现在该部分的代码已被修改。\n","    !pip install -q condacolab\n","    # Setting up condacolab and installing packages\n","    import condacolab\n","    condacolab.install_from_url(\"https://repo.anaconda.com/miniconda/Miniconda3-py39_23.11.0-2-Linux-x86_64.sh\")\n","    %cd -q /content\n","    !git clone https://github.com/RVC-Boss/GPT-SoVITS\n","\n","    %cd -q /content/GPT-SoVITS\n","\n","    !conda create -n GPTSoVits python=3.9\n","    !conda init\n","    !conda activate GPTSoVits\n","    !conda install ipykernel\n","    !bash install.sh\n","\n","    # @title Download pretrained models 下载预训练模型\n","    !mkdir -p /content/GPT-SoVITS/GPT_SoVITS/pretrained_models\n","    !mkdir -p /content/GPT-SoVITS/tools/damo_asr/models\n","    !mkdir -p /content/GPT-SoVITS/tools/uvr5\n","    %cd /content/GPT-SoVITS/GPT_SoVITS/pretrained_models\n","    !git clone https://huggingface.co/lj1995/GPT-SoVITS\n","    %cd /content/GPT-SoVITS/tools/damo_asr/models\n","    !git clone https://www.modelscope.cn/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch.git\n","    !git clone https://www.modelscope.cn/damo/speech_fsmn_vad_zh-cn-16k-common-pytorch.git\n","    !git clone https://www.modelscope.cn/damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch.git\n","    # @title UVR5 pretrains 安装uvr5模型\n","    %cd /content/GPT-SoVITS/tools/uvr5\n","    !git clone https://huggingface.co/Delik/uvr5_weights\n","    !git config core.sparseCheckout true\n","    !mv /content/GPT-SoVITS/GPT_SoVITS/pretrained_models/GPT-SoVITS/* /content/GPT-SoVITS/GPT_SoVITS/pretrained_models/\n","\n","    # 启动UI\n","    !sed -i '10s/False/True/' /content/GPT-SoVITS/config.py\n","    %cd /content/GPT-SoVITS/\n","    !/usr/local/bin/python  webui.py"],"metadata":{"id":"05lKjaKOHOWm"},"execution_count":null,"outputs":[]}]}