o
    }oi                     @   s*  d Z ddlm  mZ ddlm  mZ ddlm  m	Z
 dddZdD ]jZeejjeg d eejjedg eejjeg d eejjeg d	 eejjed
g eejjeddg eejjeg d ee
jjedg ee
jjedg ee
jjedg ee
jjedg q$ed dS )zs
WARNING: Running this test will download ALL pre-trained NeMo models.
This is bandwidth and disk space consuming.
    Nc                 C   s^   |   D ]}| j|j|d}t|| sJ q|d ur+t|tdd |   D ks-J d S d S )N)
model_namerefresh_cachec                 S   s   g | ]}|j qS  )pretrained_model_name).0mr   r   T/home/ubuntu/.local/lib/python3.10/site-packages/tests/manualtest_model_downloads.py
<listcomp>   s    z'testclass_downloads.<locals>.<listcomp>)list_available_modelsfrom_pretrainedr   
isinstanceset)clsr   model_names
model_infomodelr   r   r   testclass_downloads   s   "r   )TF)zQuartzNet15x5Base-EnzQuartzNet15x5Base-ZhzQuartzNet5x5LS-EnzQuartzNet15x5NR-EnzJasper10x5Dr-Enz!ContextNet-192-WPE-1024-8x-Stride)zMatchboxNet-3x1x64-v1zMatchboxNet-3x2x64-v1MatchboxNet-3x1x64-v2r   z!MatchboxNet-3x1x64-v2-subset-taskz!MatchboxNet-3x2x64-v2-subset-taskzMatchboxNet-VAD-3x2)speakerrecognition_speakernetspeakerverification_speakernetspeakerdiarization_speakernet
ecapa_tdnnNERModel$Punctuation_Capitalization_with_BERT*Punctuation_Capitalization_with_DistilBERT)zBERTBaseUncasedSQuADv1.1zBERTBaseUncasedSQuADv2.0zBERTLargeUncasedSQuADv1.1zBERTLargeUncasedSQuADv2.0zTacotron2-22050HzzWaveGlow-22050HzzSqueezeWave-22050HzzGlowTTS-22050Hzz+############ THAT'S ALL FOLKS! ############)N)__doc__nemo.collections.asrcollectionsasrnemo_asrnemo.collections.nlpnlpnemo_nlpnemo.collections.ttsttsnemo_ttsr   r   modelsEncDecCTCModelEncDecCTCModelBPEEncDecClassificationModelEncDecSpeakerLabelModelTokenClassificationModelPunctuationCapitalizationModelQAModelTacotron2ModelWaveGlowModelSqueezeWaveModelGlowTTSModelprintr   r   r   r   <module>   sL   
