
    oi*	                         S SK Jr  S SKJr  S SKJr  S SK	J
r  S SKJr  S SKJr  S SKJrJrJr  SqS	 r      SS\S\S\S\S\\   S\S\\   4S jjrg
)   )register_deepseek_models)register_gemma_models)register_llama_models)register_mistral_models)register_phi_models)register_qwen_models)MODEL_REGISTRY	ModelInfo	QuantTypeFc                      [         (       a  g [        5         [        5         [        5         [	        5         [        5         [        5         Sq g )NT)_ARE_MODELS_REGISTERED_register_deepseek_models_register_gemma_models_register_llama_models_register_mistral_models_register_phi_models_register_qwen_models     S/home/james-whalen/.local/lib/python3.13/site-packages/unsloth/registry/__init__.pyregister_modelsr      s7     !r   Norg	base_nameversionsizequant_typessearch_patternreturnc                   ^ [         (       d
  [        5         [        R                  " 5       nU (       a#  U Vs/ s H  owR                  U :X  d  M  UPM     nnU(       a$  U Vs/ s H  nUR
                  U:X  d  M  UPM     nnU(       a#  U Vs/ s H  owR                  U:X  d  M  UPM     nnU(       a#  U Vs/ s H  owR                  U:X  d  M  UPM     nnU(       a/  U V^s/ s H!  m[        U4S jU 5       5      (       d  M  TPM#     nnU(       a#  U Vs/ s H  nXWR                  ;   d  M  UPM     nnU$ s  snf s  snf s  snf s  snf s  snf s  snf )z
Get model info from the registry.

See registry.ModelInfo for more fields.

If search_pattern is provided, the full model path will be matched against the pattern, where the model path is the model_id on huggingface hub.

c              3   B   >#    U  H  nTR                   U:H  v   M     g 7f)N)
quant_type).0r!   
model_infos     r   	<genexpr> search_models.<locals>.<genexpr>E   s     U::((J6s   )
r   r   r	   valuesr   r   r   r   any
model_path)r   r   r   r   r   r   model_infosr#   s          `r   search_modelsr*      s`     "! '')K
)4
)4:#8MJ 	 
  *
)
##y0 ) 	 

 )4
)4:8J8Jg8UJ 	 
 )4
)4:48OJ 	 
  *
)
UUU ) 	 

  *
)
!6!66 ) 	 
 =





sG   D7D7!D<9D<E#E6EE!EEE-E)NNNNNN)	_deepseekr   r   _gemmar   r   _llamar   r   _mistralr   r   _phir   r   _qwenr   r   registryr	   r
   r   r   r   strlistr*   r   r   r   <module>r4      s    L C C I = @ : : "  #'3	33 3 	3
 i3 3 
)_3r   