+ TV-j:'^RIt^RIt^RIt^RIHt^RIHt^RIHtH t H t H t H t ^RI Ht^RIHt^RIHt^RIHtHtHtHt/RRbRRbR R bR R bR R bR R bRR bRRbRRbRRbRRbRRbRRbRRbRRbRRbRRbRRRRRRRR R!R!R"R"R#R#R$R%R&R%R'R(/ Ct^t.R7OtR)R*ltR+R,ltR8R-R.lltR8R/R0lltR9R1R2llt R3R4lt!R:R5R6llt"R#);N)Path)dedent)AnyListOptionalTupleUnion) tree_flatten)base_load_modelget_model_classget_model_path load_config qwen3_ttsoutettssparkmarvissesamecsmvoxcpmz voxcpm1.5voxcpm2vibevoice_streaming vibevoicechatterbox_turbosoprano bailingmmkitten kitten_ttsecho_ttsfish_qwen3_omni irodori_tts voxtral_tts kugelaudioaudioditlongcat_audioditlongcatz dramabox-ttsdramabox omnivoicemelotts moss_tts_nanomoss_tts_delaymoss_ttsmoss_tts_localhiggs_multimodal_qwen3higgs_audio_v3c:V^8dQhR\\,/#)return)rstr)formats"d/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_audio/tts/utils.py __annotate__r55sd3ic|\\4PR, p.pVP4'dVP 4'dnVP 4FYpVP 4'gKVP PR4'dK>VPVP 4K[ V#)z Get a list of all available TTS model types by scanning the models directory. Returns: List[str]: A list of available model type names models__) r__file__parentexistsis_diriterdirname startswithappend) models_diravailable_modelsitems r4get_available_modelsrE5sh&&1Jz0022&&(D{{}}TYY%9%9$%?%? '' 2) r6ctV^8dQhR\R\\,R\\\3,/#)r0 model_type model_namer1)r2rrr)r3s"r4r5r5Gs-3DI%S/r6c(\VVR\R7#)a Retrieve the model architecture module based on the model type and name. This function attempts to find the appropriate model architecture by: 1. Checking if the model_type is directly in the MODEL_REMAPPING dictionary 2. Looking for partial matches in segments of the model_name Args: model_type (str): The type of model to load (e.g., "outetts"). model_name (List[str]): List of model name components that might contain remapping information. Returns: Tuple[module, str]: A tuple containing: - The imported architecture module - The resolved model_type string after remapping Raises: ValueError: If the model type is not supported (module import fails). tts)rGrHcategorymodel_remapping)r MODEL_REMAPPING)rGrHs&&r4get_model_and_argsrNGs* '  r6c hV^8dQhR\R\R\R\R\P/#r0 model_pathlazystrictkwargsr1)rboolrnnModule)r3s"r4r5r5ds>    YY r6c 4\RRVRRR\RVRV/VB#)a Load and initialize the model from a given path. Args: model_path (Path): The path to load the model from. lazy (bool): If False eval the model parameters to make sure they are loaded in memory before returning, otherwise they will be loaded when needed. Default: ``False`` Returns: nn.Module: The loaded and initialized model. Raises: FileNotFoundError: If the weight files (.safetensors) are not found. ValueError: If the model class or args class are not found or cannot be instantiated. rQrKrJrLrRrS)r rMrQrRrSrTs&&&,r4 load_modelr[dsI,    (         r6c V^8dQhR\\\3,R\R\R\R\ P /#rP)r r2rrUrrVrW)r3s"r4r5r5sNFFc4i F F F F YY Fr6c "\V3RVRV/VB#)a3 Load a text-to-speech model from a local path or HuggingFace repository. This is the main entry point for loading TTS models. It automatically detects the model type and initializes the appropriate model class. Args: model_path: The local path or HuggingFace repo ID to load from. lazy: If False, evaluate model parameters immediately. strict: If True, raise an error if any weights are missing. **kwargs: Additional keyword arguments such as `revision` and `force_download`. Returns: nn.Module: The loaded and initialized model. rRrS)r[rZs&&&,r4loadr^s. j Et EF Ef EEr6c ~V^8dQhR\R\R\R\\P \ 3,/#)r0rQrRrTr1)rrUrrrVrWdict)r3s"r4r5r5s9 47 299d?r6c 8\W3/VBp\V3/VBpW43#N)r[r)rQrRrTmodelconfigs&&, r4fetch_from_hubres* z 26 2E  .v .F =r6c<V^8dQhR\R\R\/#)r0path upload_repohf_path)r2)r3s"r4r5r5s&,Y,Y,Y#,Y,Yr6c^RIp^RIHpHpHp^RIHpVPV4pVPPfR.MVPPR.,VPn \RV RV RV RV R V R V R 2 4Vn VPVPPVR 44VP!4V!4p V P!VR R7V P#VVRR7\%RV R24R#)z Uploads the model to Hugging Face hub. Args: path (str): Local path to the model. upload_repo (str): Name of the HF repo to upload to. hf_path (str): Path to the original Hugging Face model. N)HfApi ModelCardlogging) __version__mlxz # z7 This model was converted to MLX format from [`z`](https://huggingface.co/z) using mlx-audio version **zF**. Refer to the [original model card](https://huggingface.co/z) for more details on the model. ## Use with mlx ```bash pip install -U mlx-audio ``` ```bash python -m mlx_audio.tts.generate --model z3 --text "Describe this image." ``` z README.mdT)repo_idexist_okrc) folder_pathrp repo_typez0Upload successful, go to https://huggingface.co/z for details.)oshuggingface_hubrkrlrmversionrnr^datatagsrtextsavergjoinset_verbosity_info create_repo upload_folderprint) rgrhrirtrkrlrmrncardapis &&& r4 upload_to_hubrs99% >>' "D $  6eWDIINNeWi?YZaYbb~@KLLCCJ)L23>?  DI  IIbggll4-.  'COOK$O7  <[M WXr6cV^8dQhR\R\R\R\\,R\\,R\R\R\\,R \R \R \\,R \/ #) r0rimlx_pathquantize q_group_sizeq_bitsdtyperhrevision dequantizetrust_remote_codequant_predicateq_mode)r2rUrint)r3s"r4r5r5so6o6 o6o6o63- o6 SM o6  o6o6smo6o6o6c]o6 o6r6c aaa^RIHp ^RIHp HpHpHp\R4\WR7p\VRV R7wpp\V \4'd V !V V4p \VRR4oV3R loV fSp M V oVV3R lp \\VP444pVfVP!R R 4pV\"9dQ\R V4\\$V4pVP'4UUu/uFwppVVP)V4bK pppV'dV'd \+R4hV'dE\R4VP-\/VP'444V!VVVVV V R7wppV'd6\R4V !V4p\\VP444p\V\4'd \1V4pVP3RRR7RFp\4P4!\VV, 44pVFp\6P8!VV4K \4P4!\VR, V, 4RR7pVFZp\1V4P;V4pVVP<, pVP3RRR7\6P8!VV4K\ K V!VVRR7V!VVR, R7Ve\?WV4R #R #uuppi)r)mixed_quant_predicate_builder)dequantize_modelquantize_model save_config save_modelz[INFO] Loading)rT)rRrmodel_quant_predicatecR#)TrY)pms&&r4convert..sQUr6c<\VR4;'dLVPPR,^@,^8H;'d\VR4;'d S!W4#)weight to_quantized)hasattrrshape)rrrs&&r4base_quant_requirements(convert..base_quant_requirementss^ Ax  , ,r"R'1, , ,>* , ,&a+  r6Nc4<S!W4;'d S!W4#rbrY)rrroriginal_predicates&&r4rrs #A ) F F.@.F Fr6 torch_dtypez[INFO] Using dtype:z/Choose either quantize or dequantize, not both.z[INFO] Quantizing)moderz[INFO] Dequantizing)parentsrqz**) recursive) donate_modelz config.json) config_path)z*.pyz*.jsonz*.wavz*.ptz *.safetensorsz*.yamlz*.txtz*.jinja) mlx_lm.convertr mlx_lm.utilsrrrrrr re isinstancer2getattrr`r parametersgetMODEL_CONVERSION_DTYPESmxitemsastype ValueError load_weightslistrmkdirglobshutilcopy relative_tor;r) rirrrrrrhrrrrrrrrrrrQrcrdweightskvpatternfilesfile subdir_filesrel_pathdest_dirrrrs &&&&&&&&&&&& @@@r4convertrs=VV ;J"1BME6/3''7O$E+BDUV 1, < 0 0 234G } =$/ '' #U+E"29--/B/$!Q1ahhuo%/BJJKK !" 4 01(    +   #$ '|E$4$4$678(C  > NN4$N/  #j7234D KKh 'yyZ$%6%@!ATR  DDz--j9H(//1H NN4$N 7 KKh ' !! .xT2H}$<=hW5qCs3K:)float16bfloat16float32)FT)F) mlx_modelFNNNNNFTNaffine)#rrmrpathlibrtextwraprtypingrrrrr mlx.corecorermlx.nnrV mlx.utilsr mlx_audio.utilsr r r rrMMAX_FILE_SIZE_GBrrErNr[r^rerrrYr6r4rsv  44" y W h   8   h y;*y l ( =!"=#$," !J y_jj.7:<$:@F4,Y^o6o6r6