+
    TV-j£  ã                  óŽ   € ^ RI Ht ^ RIt^ RIt^ RIHt ^RIHtHt ^RI	H
t
Ht RR R lltRR R	 llt]R
8X  d
   ]! 4        R# R# )é    )ÚannotationsN)ÚOptional)ÚSUPPORTED_METRICSÚrun_seed_tts_eval)ÚDEFAULT_DATASET_REPOÚDEFAULT_REVISIONc               ó   € V ^8„  d   QhRR/# ©é   ÚargvzOptional[list[str]]© )Úformats   "Úg/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_audio/stt/eval/cli.pyÚ__annotate__r      s   € ÷ H#ñ H#Ð(ñ H#ó    c                óª  € \         P                  ! R R7      pVP                  RRRR7       VP                  R\        RR7       VP                  R\        R	R7       VP                  R
RRR7       VP                  R\
        RRR7       VP                  RRRR7       VP                  R\        P                  RRR7       VP                  RRRR7       VP                  RRRR7       VP                  RRRR7       VP                  R\
        ^RR7       VP                  R R!R".\        R#R$7       VP                  R%RR&R7       VP                  V 4      # )'z7Evaluate an STT model on a dataset with chosen metrics.)Údescriptionz--modelz$mlx-community/whisper-large-v3-turboz$STT model path or Hugging Face repo.)ÚdefaultÚhelpz--dataset-repoz9Hugging Face dataset repo containing Seed-TTS eval audio.z
--revisionzDataset revision to stream.z--output-dirTz7Directory for predictions, summaries, and cached audio.)Úrequiredr   z--limitNz.Evaluate only the first N English target wavs.)Útyper   r   z
--languagezFOptional model language hint. If omitted, no language kwarg is passed.z--gen-kwargszDAdditional model.generate kwargs as JSON, e.g. '{"temperature": 0}'.z--skip-existingÚ
store_truez6Reuse utterances already present in predictions.jsonl.)Úactionr   z--allow-missing-audioz<Skip streamed audio rows that do not have matching metadata.z--audio-cache-dirzNOptional directory for streamed wav cache. Defaults to output-dir/audio-cache.z--dataset-batch-sizezENumber of streamed dataset samples to materialize before processing. z	--metricsÚ+Úwerz-Metrics to compute. Currently supported: wer.)Únargsr   Úchoicesr   z	--verbosez!Pass verbose=True when supported.)
ÚargparseÚArgumentParserÚadd_argumentr   r   ÚintÚjsonÚloadsr   Ú
parse_args)r   Úparsers   & r   r$   r$      sÂ  € Ü×$Ò$ØMô€Fð ×ÑØØ6Ø3ð ô ð
 ×ÑØÜ$ØHð ô ð
 ×ÑØÜ Ø*ð ô ð
 ×ÑØØØFð ô ð
 ×ÑØÜØØ=ð	 ô ð ×ÑØØØUð ô ð
 ×ÑØÜ�Z‰ZØØUð	 ô ð ×ÑØØØEð ô ð
 ×ÑØØØKð ô ð
 ×ÑØØØ]ð ô ð
 ×ÑØÜØØUð	 ô ð ×ÑØØØ�Ü!Ø<ð ô ð ×ÑØ˜LÐ/Rð ô ð ×Ñ˜TÓ"Ð"r   c               ó   € V ^8„  d   QhRR/# r
   r   )r   s   "r   r   r   V   s   € ÷ =ñ =Ð"ñ =r   c                óž  € \        V 4      p\        VP                  VP                  VP                  VP
                  VP                  VP                  VP                  VP                  VP                  '       * VP                  VP                  VP                  VP                  R 7      p\        \         P"                  ! VR^R7      4       R# ))ÚmodelÚ
output_dirÚdataset_repoÚrevisionÚlimitÚlanguageÚ
gen_kwargsÚskip_existingÚfail_on_missing_audioÚaudio_cache_dirÚdataset_batch_sizeÚmetricsÚverboseF)Úensure_asciiÚindentN)r$   r   r(   r)   r*   r+   r,   r-   r.   r/   Úallow_missing_audior1   r2   r3   r4   Úprintr"   Údumps)r   ÚargsÚsummarys   &  r   Úmainr<   V   s•   € Ü�dÓ€DÜØ�j‰jØ—?‘?Ø×&Ñ&Ø—‘Ø�j‰jØ—‘Ø—?‘?Ø×(Ñ(Ø"&×":Ñ":Ô:Ø×,Ñ,Ø×2Ñ2Ø—‘Ø—‘ô€Gô 
Œ$�*Š*�W¨5¸Ô
;Ö<r   Ú__main__)N)Ú
__future__r   r   r"   Útypingr   Úrunnerr   r   Úseed_ttsr   r   r$   r<   Ú__name__r   r   r   Ú<module>rC      s7   ðÝ "ã Û Ý ç 8ß <÷H#÷V=ð( ˆzÔÙ†Fñ r   