o je@szddlZddlmZddlmZejejddddZej deZ d Z d d d d ddddZ dddZ e fddZ dS)N)request)tqdm~z.cachetortoisemodelsTORTOISE_MODELS_DIRz/data/speech_synth/models/zVhttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/autoregressive.pthzRhttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/classifier.pthzMhttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/clvp2.pthzYhttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/diffusion_decoder.pthzOhttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/vocoder.pthzPhttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/rlg_auto.pthzThttps://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/rlg_diffuser.pth)zautoregressive.pthzclassifier.pthz clvp2.pthzdiffusion_decoder.pthz vocoder.pthz rlg_auto.pthzrlg_diffuser.pthc CstjtddtD]M\}}|dur||vrq tjt|}tj|r&q td|d|dt dddd d }t |||fd d Wdn1sOwYtd q dS)z= Call to download all the models that Tortoise uses. T)exist_okNz Downloading z from z...Bi)unit unit_scale unit_divisorminiterscSs||||jSN)updaten)nbbsfstrP/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/layers/tortoise/utils.py!sz!download_models..zDone.) osmakedirs MODELS_DIRMODELSitemspathjoinexistsprintrr urlretrieve)specific_models model_nameurl model_pathrrrrdownload_modelss  r'cCsH|tvr td|dtj||}tj|s"|tkr"t|g|S)zC Get path to given model, download it if it doesn't exist. zModel z not found in available models.)r ValueErrorrrrr rr')r$ models_dirr&rrrget_model_path%s  r*r)rurllibrrrr expanduserDEFAULT_MODELS_DIRenvirongetrrr'r*rrrrs