+ i\i:,^RIt^RIt^RIt^RIt^RIt^RIHt^RIHt^RI H t H t H t H t ^RIHt^RIHt^RIHt^RIHtRRRRR R R R R R R R RRRRRR/ t^t.R!OtR"RRlltRtRRltRRltR#RRlltR$RRlltRRlt R%RR llt!R#)&N)Path)dedent)ListOptionalTupleUnion)snapshot_download) tree_flattenoutettssparkmarvissesamecsmvoxcpmz voxcpm1.5vibevoice_streaming vibevoicechatterbox_turbosopranocRV^8dQhR\R\\,R\/#)path_or_hf_reporevisionreturn)strrr)formats"k/Users/agent/.cache/uv/archive-v0/6ZL4-8mCcRNgGE9wi6M6J/lib/python3.14/site-packages/mlx_audio/tts/utils.py __annotate__rs%##C#8C=#D#cz\V4pVP4'g\\VV.ROR74pV#)ab Ensures the model is available locally. If the path does not exist locally, it is downloaded from the Hugging Face Hub. Args: path_or_hf_repo (str): The local path or Hugging Face repository ID of the model. revision (str, optional): A revision id which can be a branch name, a tag, or a commit hash. Returns: Path: The path to the model. )rallow_patterns) *.json *.safetensors*.pyz*.modelz *.tiktoken*.txtz*.jsonl*.yaml*.wavr$z*.pth)rexistsr )rr model_paths&& rget_model_pathr)sBo&J      !   ( rc|\\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__parentr'is_diriterdirname startswithappend) models_diravailable_modelsitems rget_available_modelsr7Esh&&1Jz0022&&(D{{}}TYY%9%9$%?%? '' 2) rcFV^8dQhR\R\\,/#)r model_type model_name)rr)rs"rrrWs,,3,DI,rcR\PVR4p\4pVe2W 8wd,VF%pWC9dTpV\9gK\V,pM \P!RV 24pWP3# \ d)RT R2p\ P!T4\T4hi;i)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). Nzmlx_audio.tts.models.z Model type z not supported.) MODEL_REMAPPINGgetr7 importlib import_module ImportErrorloggingerror ValueError)r9r:model_type_mappedr+partarchmsgs&& rget_model_and_argsrHWs,(++J=" #F"3"AD~! &,T2 &&)>zl'KL   J<7 cos A333B&cRV^8dQhR\\\3,R\/#)rr(r)rrrdict)rs"rrrs'RRE#t),R4Rrc ^RIHp\V\4'd \ V4pVP !V3/VBP 4# \dy\TR, RR7;_uu_4p\P!T4uuRRR4u# +'giR#;i \dp\RT 24ThRp?ii;ii;i)aALoad model configuration from a path or Hugging Face repo. Args: model_path: Local path or Hugging Face repo ID to load config from **kwargs: Additional keyword arguments to pass to the config loader Returns: dict: Model configuration Raises: FileNotFoundError: If config.json is not found at the path ) AutoConfig config.jsonzutf-8)encodingNzConfig not found at ) transformersrL isinstancerr)from_pretrainedto_dictrCopenjsonloadFileNotFoundError)r(kwargsrLfexcs&, r load_configrZs(*c""#J/ R))*??GGII R Rj=07CCqyy|DCCC  R#&::,$GHc Q R RsM A C B*2B B*C  B' B*'B** C 5CC  C c\V^8dQhR\R\R\R\P/#)rr(lazystrictr)rboolnnModule)rs"rrrs6GGG G26GYYGrc aaRp\V\4'dAVP4PR4R,PR4p\ V4p\V\ 4'dkVP PR4pVP V^,,P4PR4R,PR4pM\R\V4 24h\V3/VBoVSR&\V4SR&SPR R4pVfSPR R4pVfVeV^,P4MRpSPR R4p\P!\VR , 44pV'g.\P!\VR , R , 44pV'g+\P!RV 24RV R2p \!V 4h/oVF(p SP#\$P&!V 44K* \)WdR7wr\+V R4'dV P,P/S4MSp V e\+V R4'dW nV P3V 4p SPR R4pVfV P5S4oSPR R4;pe0VV3Rlp\6P8!V VR,VR,VR7V P;\=SP?44VR7V'g%\$P@!V PC44V PA4\+V P2R4'dV P2PEW4p V # \d.TPP4PR4pELi;i)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. N/-hubz--zInvalid model path type: tokenizer_namer(r9 architecture quantizationr"LLMzNo safetensors found in z No safetensors found in a Create safetensors using the following code: ``` from transformers import AutoModelForCausalLM, AutoProcessor model_id= "" model = AutoModelForCausalLM.from_pretrained(model_id) processor = AutoProcessor.from_pretrained(model_id) model.save_pretrained("") processor.save_pretrained("") ``` Then use the as the --hf-path in the convert script. ``` python -m mlx_audio.tts.convert --hf-path --mlx-path ``` )r9r: ModelConfigc<VSR,9dSR,V,#\VR4'gR#\VR4'd%VPP^@,^8wdR#V R2S9#)rg to_quantizedFweightz.scales)hasattrrlsize)pmconfigweightss&&rget_class_predicate'load_model..get_class_predicate seF>**n-a001n--q(## (:a(?S=G+ +r group_sizebits)rurvclass_predicate)r]post_load_hook)#rPrlowersplitr)rpartsindexrCr1typerZr=globrArBrVupdatemxrUrHrmri from_dictr(Modelsanitizer_quantize load_weightslistitemseval parametersrx)r(r\r]rWr:r}r9rg weight_filesmessagewf model_class model_configmodelrsrqrrs&&&, @@r load_modelrsM&J*c""%%'--c226<?L yyZ%%7/%I!JK  0 =># %  $ ((Grwwr{#1K ; . . ))&1  GL,$G$G",   l +E::nd3L..) >488 E , #L1f%/   tGMMO,V<    "# JJL {  "233!!00C LY <#..066s;J >' "D $  6eWDIINNeWi?YZaYbb~@KLLCCJ)L23>?  DI  IIbggll4-.  'COOK$O7  <[M WXrcV^8dQhR\R\R\R\R\R\R\R\\,R \R \R \\,/ #) rrmlx_pathr q_group_sizeq_bitsdtyperr dequantizetrust_remote_codequant_predicate)rr^intr)rs"rrrcsh6h6 h6h6h6 h6  h6  h6h6smh6h6h6c]h6rc aaa^RIHp ^RIHp Hp HpHp\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'dC\R4VP-\/VP'444V !VVW4V 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)r\rmodel_quant_predicatecR#)T)rorps&&rconvert..}sQUrc<\VR4;'dLVPPR,^@,^8H;'d\VR4;'d S!W4#)rlrkry)rmrlshape)rorprs&&rbase_quant_requirements(convert..base_quant_requirementss^ Ax  , ,r"R'1, , ,>* , ,&a+  rNc4<S!W4;'d S!W4#rr)rorproriginal_predicates&&rrrs #A ) F F.@.F Fr torch_dtypez[INFO] Using dtype:z/Choose either quantize or dequantize, not both.z[INFO] Quantizing)rz[INFO] Dequantizing)parentsrz**) recursive) donate_modelrM) config_path)r#r!r&z*.ptr"r%r$) mlx_lm.convertr mlx_lm.utilsrrrrrr)rrPrgetattrrJr rr=MODEL_CONVERSION_DTYPESrrastyperCrrrmkdirrshutilcopy relative_tor.r)rrrrrrrrrrrrrrrrr(rrqrrkvpatternfilesfile subdir_filesrel_pathdest_dirrrrs&&&&&&&&&&& @@@rconvertrcs=VV ;J"1BME6/3''7O$E+BDUV 1, < 0 0 234G } =$/ '' #U+E"29--/B/$!Q1ahhuo%/BJJKK !" 4 01( 6<  #$ '|E$4$4$678(C  > NN4$N/ #j7234D KKh 'yyZ$%6%@!ATR  DDz--j9H(//1H NN4$N 7 KKh ' !,xT2H}$<=hW5eCs3K8)float16bfloat16float32r)FT)F) mlx_modelF@NNNFTN)"rr>rTrArpathlibrtextwraprtypingrrrrmlx.corecorermlx.nnr_rr mlx.utilsr r<MAX_FILE_SIZE_GBrr)r7rHrZrrrrrrrrs   //-"y W h 8 h;* y <#N$,^R8GT,Y^h6h6r