Merge pull request #188 from magic-akari/4.0-Vec768-Layer12

fix(colab): correct `diff_model_URL`
This commit is contained in:
YuriHead 2023-05-24 00:20:23 +08:00 committed by GitHub
commit 8d2339ce5a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 1 deletions

View File

@ -176,7 +176,7 @@
" !curl -L {G_0_URL} -o logs/44k/G_0.pth\n", " !curl -L {G_0_URL} -o logs/44k/G_0.pth\n",
"\n", "\n",
"if download_pretrained_diffusion_model:\n", "if download_pretrained_diffusion_model:\n",
" !curl -L {diffusion_URL} -o logs/44k/diffusion/model_0.pt" " !curl -L {diff_model_URL} -o logs/44k/diffusion/model_0.pt"
] ]
}, },
{ {
@ -460,6 +460,21 @@
"#@markdown\n", "#@markdown\n",
"\n", "\n",
"%cd /content/so-vits-svc\n", "%cd /content/so-vits-svc\n",
"\n",
"import os\n",
"if not os.path.exists(\"./pretrain/nsf_hifigan/model\"):\n",
" !curl -L https://github.com/openvpi/vocoders/releases/download/nsf-hifigan-v1/nsf_hifigan_20221211.zip -o nsf_hifigan_20221211.zip\n",
" !unzip nsf_hifigan_20221211.zip\n",
" !rm -rf pretrain/nsf_hifigan\n",
" !mv -v nsf_hifigan pretrain\n",
"\n",
"#@markdown Whether to enable tensorboard\n",
"tensorboard_on = True #@param {type:\"boolean\"}\n",
"\n",
"if tensorboard_on:\n",
" %load_ext tensorboard\n",
" %tensorboard --logdir logs/44k\n",
"\n",
"!python train_diff.py -c configs/diffusion.yaml" "!python train_diff.py -c configs/diffusion.yaml"
] ]
}, },