
    eTh                         S SK Jr  S SKrS SKJrJrJr  SSKJrJ	r	J
r
  \
R                  " \5      rS\S\S\S	\S
\S   4
S jrS	\S\S\4S jrS\S\4S jrSS\4S jjrg)    )OptionalN)
DiscussionHfApiget_repo_discussions   )cached_filehttp_user_agentloggingapimodel_idpr_titletokenreturnr   c                 *   U R                  XS9S   R                  n[        XS9 Hj  nUR                  U:X  d  M  UR                  S:X  d  M'  UR
                  (       d  M:  U R                  XR                  US9nXFS   R                  :X  d  Mh  Us  $    g )Nr   r   )repo_idr   openrevisionr   r   )list_repo_commits	commit_idr   titlestatusis_pull_requestgit_reference)r   r   r   r   main_commit
discussioncommitss          [/var/www/auris/envauris/lib/python3.13/site-packages/transformers/safetensors_conversion.pyprevious_prr       s    '''>qAKKK*8I
x'J,=,=,GJLfLfLf++H?W?W_d+eGaj222!! J     privatec                    [         R                  S5        SnU S3nS nSX!U /0n[        R                  " USUS9R	                  5       nUS   n[        R
                  " U S	U 3SS
9 n	 [         R                  S5        U" U	5        S S S 5        g ! [         a+  n
[         R                  S[        U
5       35         S n
A
N9S n
A
ff = f! , (       d  f       g = f)Nz;Attempting to convert .bin model on the fly to safetensors.z$https://safetensors-convert.hf.spacez	/call/runc                 `   U R                  5        H  nUR                  5       nUR                  S5      (       a\  USS  n[        R	                  SU 35        US:X  a    g US:X  a  [        R	                  S5        Mk  [        R	                  SU 35        M  [        R	                  U5        M     g )Nzevent:   zSafetensors conversion status: complete	heartbeat	HeartbeatzUnknown status )
iter_linesdecode
startswithloggerdebug)_sse_connectionliner   s      r   startspawn_conversion.<locals>.start   s    #..0D;;=Dx((ab>vhGHZ'{*LL-LL?6(!;<T" 1r!   dataT)streamjsonevent_id/)r3   z*Spawning safetensors automatic conversion.zError during conversion: )
r,   inforequestspostr4   getr-   	Exceptionwarningrepr)r   r"   r   safetensors_convert_space_urlsse_urlr0   r2   resultr5   sse_connectiones              r   spawn_conversionrC      s    
KKMN$J!./y9G#  X./D]]74d;@@BFj!H		8*-d	;~	BLLEF.! 
<	;  	BNN6tAwi@AA	B	 
<	;s0   )C	+B
C!C<C	CC		
Cc                 J   U R                  U5      R                  n[        R                  S5        SnUR	                  S5      n[        XXES9nUb  U(       d'  UR                  S:w  a  [        XSU5        [        XXES9nO[        R                  S5        SUR                   3nU$ )Nz(Attempting to create safetensors variantz*Adding `safetensors` variant of this modelr   r   SFconvertbotzSafetensors PR existszrefs/pr/)	
model_infor"   r,   r7   r:   r    authorrC   num)r   r   kwargsr"   r   r   prshas           r   get_conversion_pr_referencerL   :   s    nnX&..G
KK:;;HJJwE
 
SH	:B	z'bii>&A2>+,RVVH
CJr!   pretrained_model_name_or_pathc                 0    [        UR                  S5      S[        5       0S9n[        X040 UD6nUc  gXBS'   US	 UR	                  U SUUR                  S5      S9nU(       a  SOS	n[        X40 UD6nXtU4$ ! [         a  nU(       d  Ue S nAg S nAff = f)
Nr   z
user-agent)r   headers)NNr   _commit_hashzmodel.safetensors.index.jsonr   zmodel.safetensors)r   r:   r	   rL   file_existsr   r;   )	rM   ignore_errors_during_conversioncached_file_kwargsr   rK   shardedfilenameresolved_archive_filerB   s	            r   auto_conversionrW   Q   s    ,009LRaRcCde)#cPbc;),:&~. //)*$((1	 " 
 6=1BU +,I jWi j$722 .G /s   2A8 AA8 8
B	BB)F)typingr   r8   huggingface_hubr   r   r   utilsr   r	   r
   
get_logger__name__r,   strr    boolrC   rL   rW    r!   r   <module>r`      s      C C 8 8 
		H	%U c S  R^I_  BC  B$  B#  BFU c .3 r!   