o 㥵i @sddlmZddlZddlmZddlZejjZejZdddddZ d d d d d Z ej dddd dde de de fddZ    dde de de de fddZdS))PathN)ml0.0.10.0.40.0.51.0.0))44khz8kbps)24khzr )16khzr )r16kbpszWhttps://github.com/descriptinc/descript-audio-codec/releases/download/0.0.1/weights.pthz]https://github.com/descriptinc/descript-audio-codec/releases/download/0.0.4/weights_24khz.pthz]https://github.com/descriptinc/descript-audio-codec/releases/download/0.0.5/weights_16khz.pthzdhttps://github.com/descriptinc/descript-audio-codec/releases/download/1.0.0/weights_44khz_16kbps.pth))rrr )r rr )r rr )rrr downloadT)group positionalwithout_prefixrr latest model_type model_bitratetagcCs|}|}|dvsJd|dvsJd|dkr"t||f}t|||fd}|dur9td|d|td d d d |d |d |d}|sw|jj dddddl }||}|j dkrqtd|j | |j |S)a> Function that downloads the weights file from URL if a local cache is not found. Parameters ---------- model_type : str The type of model to download. Must be one of "44khz", "24khz", or "16khz". Defaults to "44khz". model_bitrate: str Bitrate of the model. Must be one of "8kbps", or "16kbps". Defaults to "8kbps". Only 44khz model supports 16kbps. tag : str The tag of the model to download. Defaults to "latest". Returns ------- Path Directory path required to load model via audiotools. )rr r z6model_type must be one of '44khz', '24khz', or '16khz')r r z1model_bitrate must be one of '8kbps', or '16kbps'rNzCould not find model with tag z and model type z.cachedescriptdacweights__z.pthT)parentsexist_okrz1Could not download model. Received response code )lower__MODEL_LATEST_TAGS____MODEL_URLS__get ValueErrorrhomeexistsparentmkdirrequests status_code write_bytescontent)rrr download_link local_pathr%responser,F/home/ubuntu/.local/lib/python3.10/site-packages/dac/utils/__init__.pyr *sF        load_pathcCs |s t|||d}t|}|S)N)rrr)r DACload)rrrr. generatorr,r,r- load_modelps  r2)rr r)rr rN)pathlibrargbind audiotoolsrrmodelr/ Acceleratorrrbindstrr r2r,r,r,r-sL    F