fix(colab): download hifigan if not exist

This commit is contained in:
magic-akari 2023-05-23 20:28:01 +08:00
parent aa40374526
commit cf6a46e99d
No known key found for this signature in database
GPG Key ID: EC005B1159285BDD
1 changed files with 15 additions and 0 deletions

View File

@ -460,6 +460,21 @@
"#@markdown\n",
"\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"
]
},