
    Z j                         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\S\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          t/root/GenerationalWealth/GenerationalWealth/venv/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
                  " S	U S
U 35       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[        5      (       d  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        M  [
        R                  SU 35        M  [
        R                  U5        M     g )Nzevent:   zSafetensors conversion status: complete	heartbeat	HeartbeatzUnknown status )
iter_lines
isinstancestrdecode
startswithloggerdebug)_sse_connectionliner   s      r   startspawn_conversion.<locals>.start   s    #..0DdC(({{}x((ab>vhGHZ'{*LL-LL?6(!;<T" 1r!   dataT)follow_redirectsjsonevent_idGET/z*Spawning safetensors automatic conversion.zError during conversion: )
r.   infohttpxpostr6   streamr/   	Exceptionwarningrepr)r   r"   r   safetensors_convert_space_urlsse_urlr2   r4   resultr7   sse_connectiones              r   spawn_conversionrF      s    
KKMN$J!./y9G#" X./DZZ$TBGGIFj!H	ey(4	5	BLLEF.! 
6	5  	BNN6tAwi@AA	B	 
6	5s0   +C-B
C!C>CCC
Cc                 f   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        Uc  [        S5      eSUR                   3nU$ )	Nz(Attempting to create safetensors variantz*Adding `safetensors` variant of this modelr   r   SFconvertbotzSafetensors PR existszCould not create safetensors conversion PR. The repo does not appear to have a file named pytorch_model.bin or model.safetensors.If you are loading with variant, use `use_safetensors=False` to load the original model.zrefs/pr/)

model_infor"   r.   r:   getr    authorrF   OSErrornum)r   r   kwargsr"   r   r   prshas           r   get_conversion_pr_referencerQ   ;   s    nnX&..G
KK:;;HJJwE
 
SH	:B	z'bii>&A2>+,	zg
 	

 RVVH
CJr!   pretrained_model_name_or_pathignore_errors_during_conversionsafe_weights_namesafe_weights_index_namec                 0    [        UR                  S5      S[        5       0S9n[        XP40 UD6nUc  gXdS'   US	 UR	                  U UUUR                  S5      S9nU(       a  UOUn[        X40 UD6n	XU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_hashr   )r   rJ   r	   rQ   file_existsr   r>   )rR   rS   rT   rU   cached_file_kwargsr   rP   shardedfilenameresolved_archive_filerE   s              r   auto_conversionr^   W   s    ,009LRaRcCde)#cPbc;),:&~. //)#$((1	 " 
 /6*;L +,I jWi j$722 .G /s   2A8 AA8 8
B	BB)Fzmodel.safetensorszmodel.safetensors.index.json)typingr   r;   huggingface_hubr   r   r   utilsr   r	   r
   
get_logger__name__r.   r+   r    boolrF   rQ   r^    r!   r   <module>rf      s      C C 8 8 
		H	%U c S  R^I_ !BC !B$ !B# !BHU c < -20#A	#&%)  !	r!   