o
    }o™i  ã                   @   sÐ  d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZd d dlemfZfmgZg d dlhmiZimjZj d dlkmlZl g d¢ZmdS )é    )dÚbaichuan2_7bÚ	bert_110mÚ	bert_340mÚchatglm3_6bÚdeepseek_v2Údeepseek_v2_liteÚdeepseek_v3Úe5_340mÚgemma2Ú	gemma2_2bÚ	gemma2_9bÚ
gemma2_27bÚ	gemma3_1bÚgemma_2bÚgemma_7bÚ	gpt3_175bÚhf_auto_model_for_causal_lmÚhyena_1bÚhyena_7bÚ	hyena_40bÚ
hyena_baseÚ	llama2_7bÚ	llama3_8bÚllama3_8b_16kÚllama3_8b_64kÚllama3_8b_128kÚ
llama3_70bÚllama3_70b_16kÚllama3_70b_64kÚ
llama4_e16Úllama4_e128Ú
llama31_8bÚllama31_70bÚllama31_405bÚllama31_nemotron_70bÚllama31_nemotron_nano_8bÚllama31_nemotron_ultra_253bÚ
llama32_1bÚ
llama32_3bÚllama33_nemotron_super_49bÚllama_embedding_1bÚllama_embedding_3bÚllama_reranker_1bÚmamba2_1_3bÚmamba2_2_7bÚ	mamba2_8bÚmamba2_130mÚmamba2_370mÚmamba2_780mÚmamba2_hybrid_8bÚ
mistral_7bÚmistral_nemo_12bÚmixtral_8x7bÚmixtral_8x7b_16kÚmixtral_8x7b_64kÚmixtral_8x22bÚmixtral_8x22b_64kÚnemotronÚnemotron3_4bÚnemotron3_8bÚnemotron3_22bÚnemotron3_22b_16kÚnemotron3_22b_64kÚnemotron4_15bÚnemotron4_15b_16kÚnemotron4_15b_64kÚnemotron4_340bÚnemotronh_4bÚnemotronh_8bÚnemotronh_47bÚnemotronh_56bÚphi3_mini_4k_instructÚqwen2Ú
qwen2_1p5bÚqwen2_7bÚ	qwen2_72bÚ
qwen2_500mÚqwen3Ú
qwen3_1p7bÚqwen3_4bÚqwen3_8bÚ	qwen3_14bÚqwen3_30b_a3bÚ	qwen3_32bÚqwen3_235b_a22bÚ
qwen3_600mÚqwen25_1p5bÚ	qwen25_7bÚ
qwen25_14bÚ
qwen25_32bÚ
qwen25_72bÚqwen25_500mÚ
starcoder2Ústarcoder2_3bÚstarcoder2_7bÚstarcoder2_15bÚstarcoder_15bÚt5_3bÚt5_11bÚt5_220m)Údefault_logÚdefault_resume)ÚadamÚsgd)Útorchrun)ir   r   r   r   r   r   r   r	   r   r   r   r   r   r   r   r   r   r   r!   r"   r#   r'   r(   r   r    r%   r)   r&   r$   r*   r+   r,   r0   r1   r2   r-   r.   r/   r3   rE   rF   rG   rH   r4   r5   r   r   r   r   r   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rI   re   rc   rd   rb   r^   r_   r`   ra   rJ   rN   rK   rL   rM   r]   rX   rY   rZ   r[   r\   rO   rW   rP   rQ   rR   rS   rU   rT   rV   r   r
   r   r   r   r   rh   ri   rf   rg   rj   N)nÚnemo.collections.llm.recipesr   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   Ú(nemo.collections.llm.recipes.log.defaultrf   rg   Ú"nemo.collections.llm.recipes.optimrh   ri   Ú)nemo.collections.llm.recipes.run.executorrj   Ú__all__© rp   rp   úY/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/llm/recipes/__init__.pyÚ<module>   s   þš f