
    oi                     d   S SK JrJrJrJr  SqSqSqSqSq	Sq
 " S S\5      r " S S\5      r\" SSS	/S
S/\S\R                  \R                  /S9r\" SSS	/SS/\S\R                  \R                   /S9r\" SSS	/SS/\S\R                  \R                  \R                   /S9r\" SSS	/SS/\S\R                  \R                   /S9r\" SSS	/SSS/\S\R(                  \R                   /\R                   /S.S9r\" SSS	/S/ SQ\S\R(                  \R,                  \R                   /\R(                  \R,                  /\R(                  \R,                  \R                   /\R                   \R,                  /S.S9rS&S\4S jjrS&S\4S jjrS&S\4S jjrS&S\4S jjrS&S\4S jjrS&S\4S jjrS&S\4S jjrS r \" S S!9  \!S":X  a^  S S#K J"r"J#r#  \"RH                  " 5         \" S S!9  \"RJ                  " 5        H)  u  r&r'\#" \&5      r'\'c  \(" S$\& 35        M  \(" S%\& 35        M+     g	g	)'    )	ModelInfo	ModelMeta	QuantType_register_modelsFc                   2   ^  \ rS rSr\U 4S j5       rSrU =r$ )DeepseekV3ModelInfo   c                 4   > U SU 3n[         TU ]  XX4XV5      $ )Nz-Vsuperconstruct_model_namecls	base_nameversionsize
quant_typeinstruct_tagkey	__class__s          T/home/james-whalen/.local/lib/python3.13/site-packages/unsloth/registry/_deepseek.pyr   (DeepseekV3ModelInfo.construct_model_name   s,    2gY'w+,
 	
     __name__
__module____qualname____firstlineno__classmethodr   __static_attributes____classcell__r   s   @r   r   r      s    
 
r   r   c                   2   ^  \ rS rSr\U 4S j5       rSrU =r$ )DeepseekR1ModelInfo   c                 d   > U(       a  U SU 3OUnU(       a  U SU S3n[         TU ]  XX4XV5      $ )N-Br   r   s          r   r   (DeepseekR1ModelInfo.construct_model_name   sE    *11WI&yE4&"Cw+,
 	
r   r   r   r#   s   @r   r%   r%      s    
 
r   r%   zdeepseek-aiDeepSeekN3 )orgr   instruct_tagsmodel_versionmodel_sizesmodel_info_clsis_multimodalquant_typesz3-0324zDeepSeek-R1ZerozDeepSeek-R1-DistillLlama870)r7   r8   Qwen)z1.571432include_original_modelc                 <    [         (       a  g [        [        U S9  Sq g Nr=   T)_IS_DEEPSEEK_V3_REGISTEREDr   DeepseekV3Metar@   s    r   register_deepseek_v3_modelsrC   i       !!^>TU!%r   c                 <    [         (       a  g [        [        U S9  Sq g r?   )_IS_DEEPSEEK_V3_0324_REGISTEREDr   DeepseekV3_0324Metar@   s    r    register_deepseek_v3_0324_modelsrH   q   s    &&(CYZ&*#r   c                 <    [         (       a  g [        [        U S9  Sq g r?   )_IS_DEEPSEEK_R1_REGISTEREDr   DeepseekR1Metar@   s    r   register_deepseek_r1_modelsrL   y   rD   r   c                 <    [         (       a  g [        [        U S9  Sq g r?   )_IS_DEEPSEEK_R1_ZERO_REGISTEREDr   DeepseekR1ZeroMetar@   s    r    register_deepseek_r1_zero_modelsrP      s    &&'BXY&*#r   c                 <    [         (       a  g [        [        U S9  Sq g r?   )(_IS_DEEPSEEK_R1_DISTILL_LLAMA_REGISTEREDr   DeepseekR1DistillLlamaMetar@   s    r   )register_deepseek_r1_distill_llama_modelsrT      s!    //"=S 04,r   c                 <    [         (       a  g [        [        U S9  Sq g r?   )'_IS_DEEPSEEK_R1_DISTILL_QWEN_REGISTEREDr   DeepseekR1DistillQwenMetar@   s    r   (register_deepseek_r1_distill_qwen_modelsrX      s!    ..!<R /3+r   c                 p    [        U S9  [        U S9  [        U S9  [        U S9  [	        U S9  [        U S9  g )Nr@   )rC   rH   rL   rP   rT   rX   r@   s    r   register_deepseek_modelsrZ      s>    9OP$>TU9OP$>TU-!7 -!7r   c                      SSK Jn   SSK Jn  U" SSSS9n/ nU HE  nUR                  nUR	                  S5      S	   nUR                  S
5      nUR                  U5        MG     U$ )Nr   )r   )list_modelsunslothDistilli  )authorsearchlimit/zDeepSeek-R1-Distill-)unsloth.utils.hf_hubr   r\   idsplitremoveprefixappend)HfModelInfor\   modelsdistill_modelsmodelmodel_id
model_namer   s           r    _list_deepseek_r1_distill_modelsro      sm    =0 +Y!F N88^^C(,
))*@Ag&  r   Tr@   __main__)MODEL_REGISTRY_check_model_infou   ✘ u   ✓ )F))unsloth.registry.registryr   r   r   r   rA   rF   rJ   rN   rR   rV   r   r%   NONEBF16rB   GGUFrG   rK   rO   UNSLOTHrS   BNBrW   boolrC   rH   rL   rP   rT   rX   rZ   ro   r   rq   rr   clearitemsrm   
model_infoprintr   r   r   <module>r~      s   W W" "' " "' +0 (*/ '
) 

) 
 
F$(>>9>>2	  
F$(>>9>>2	  
F$(>>9>>9>>B	 
F$(>>9>>2	  '
%F+("**INN;INNCST	  &
%F*(!!9==)..A/  )--@~~y}}-	 "& &+T +& &+T +4d 43T 3
T 
$ $ 7zKd; . 4 4 6*&x0
GH:&'GH:&' !7 r   