Ë
    S^(hê  ã                   ó¸   — d dl mZ d dlZd dlmZmZmZ ddlmZm	Z	m
Z
  e
j                  e«      Zdededed	ed
ed   f
d„Zd	ededefd„Zdedefd„Zddefd„Zy)é    )ÚOptionalN)Ú
DiscussionÚHfApiÚget_repo_discussionsé   )Úcached_fileÚhttp_user_agentÚloggingÚapiÚmodel_idÚpr_titleÚtokenÚreturnr   c                 ó*  — | j                  ||¬«      d   j                  }t        ||¬«      D ]c  }|j                  |k(  sŒ|j                  dk(  sŒ#|j
                  sŒ0| j                  ||j                  |¬«      }||d   j                  k(  sŒa|c S  y )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          úa/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/safetensors_conversion.pyÚprevious_prr       s–   € Ø×'Ñ'¨¸Ð'Ó>¸qÑA×KÑK€KÜ*°8À5ÔIò "ˆ
Ø×Ñ˜xÓ'¨J×,=Ñ,=ÀÓ,GÈJ×LfÓLfØ×+Ñ+¨H¸z×?WÑ?WÐ_dÐ+ÓeˆGà˜g a™j×2Ñ2Ó2Ø!Ò!ð"ð ó    Úprivatec                 ó²  — t         j                  d«       d}|› d�}d„ }d||| gi}t        j                  |d|¬«      j	                  «       }|d   }t        j
                  |› d	|› �d¬
«      5 }		 t         j                  d«        ||	«       d d d «       y # t        $ r+}
t         j                  dt        |
«      › �«       Y d }
~
Œ8d }
~
ww xY w# 1 sw Y   y xY w)Nz;Attempting to convert .bin model on the fly to safetensors.z$https://safetensors-convert.hf.spacez	/call/runc                 óJ  — | j                  «       D ]�  }|j                  «       }|j                  d«      rX|dd  }t        j	                  d|› �«       |dk(  r y |dk(  rt        j	                  d«       Œct        j	                  d|› �«       Œ|t        j	                  |«       Œ’ y )Nzevent:é   zSafetensors conversion status: ÚcompleteÚ	heartbeatÚ	HeartbeatzUnknown status )Ú
iter_linesÚdecodeÚ
startswithÚloggerÚdebug)Ú_sse_connectionÚliner   s      r   Ústartzspawn_conversion.<locals>.start   sŽ   € Ø#×.Ñ.Ó0ò 	#ˆDØ—;‘;“=ˆDØ�‰˜xÔ(Ø˜a˜b˜�Ü—‘Ð>¸v¸hÐGÔHà˜ZÒ'ÙØ˜{Ò*Ü—L‘L Õ-ä—L‘L ?°6°(Ð!;Õ<ä—‘˜TÕ"ñ	#r!   ÚdataT)ÚstreamÚjsonÚevent_idú/)r2   z*Spawning safetensors automatic conversion.zError during conversion: )
r,   ÚinfoÚrequestsÚpostr3   Úgetr-   Ú	ExceptionÚwarningÚrepr)r   r"   r   Úsafetensors_convert_space_urlÚsse_urlr0   r1   Úresultr4   Ússe_connectionÚes              r   Úspawn_conversionrB      sã   € Ü
‡K�KÐMÔNà$JÐ!Ø.Ð/¨yÐ9€Gò#ð  �X˜w¨Ð.Ð/€Dä�]‰]˜7¨4°dÔ;×@Ñ@ÓB€FØ�jÑ!€Hä	�‰˜˜	  8 *Ð-°dÔ	;ð B¸~ð	BÜ�L‰LÐEÔFÙ�.Ô!÷Bð Bøô ò 	BÜ�N‰NÐ6´t¸A³w°iÐ@×AÑAûð	Bú÷	Bð Bús0   Á.CÁ0BÂ	C
Â!CÃ CÃC
Ã
CÃCc                 óR  — | j                  |«      j                  }t        j                  d«       d}|j	                  d«      }t        | |||¬«      }|�|s,|j                  dk7  rt        |||«       t        | |||¬«      }nt        j                  d«       d|j                  › �}|S )Nz(Attempting to create safetensors variantz*Adding `safetensors` variant of this modelr   r   ÚSFconvertbotzSafetensors PR existszrefs/pr/)	Ú
model_infor"   r,   r6   r9   r    ÚauthorrB   Únum)r   r   Úkwargsr"   r   r   ÚprÚshas           r   Úget_conversion_pr_referencerK   :   s˜   € Ø�n‰n˜XÓ&×.Ñ.€Gä
‡K�KÐ:Ô;Ø;€HØ�J‰J�wÓ€Eô
 
�S˜( H°EÔ	:€Bà	€z™' b§i¡i°>Ò&AÜ˜ ¨Ô2Ü˜˜h¨¸Ô>‰ä�‰Ð+Ô,à�R—V‘V�HÐ
€Cà€Jr!   Úpretrained_model_name_or_pathc                 ó,  — 	 t        |j                  d«      dt        «       i¬«      }t        || fi |¤Ž}|€y||d<   |d= |j	                  | d||j                  d«      ¬«      }|rdnd	}t        | |fi |¤Ž}|||fS # t        $ r}|s|‚Y d }~y d }~ww xY w)
Nr   z
user-agent)r   Úheaders)NNr   Ú_commit_hashzmodel.safetensors.index.jsonr   zmodel.safetensors)r   r9   r	   rK   Úfile_existsr   r:   )	rL   Úignore_errors_during_conversionÚcached_file_kwargsr   rJ   ÚshardedÚfilenameÚresolved_archive_filerA   s	            r   Úauto_conversionrV   Q   sÍ   € ðÜÐ,×0Ñ0°Ó9ÀLÔRaÓRcÐCdÔeˆÜ)¨#Ð/LÑcÐPbÑcˆàˆ;ØØ),Ð˜:Ñ&Ø˜~Ð.ð —/‘/Ø)Ø*ØØ$×(Ñ(¨Ó1ð	 "ó 
ˆñ 6=Ñ1ÐBUˆä +Ð,IÈ8Ñ jÐWiÑ jÐØ$ c¨7Ð2Ð2øÜò Ù.ØˆGô /ûðús   ‚5A< ¸AA< Á<	BÂBÂB)F)Útypingr   r7   Úhuggingface_hubr   r   r   Úutilsr   r	   r
   Ú
get_loggerÚ__name__r,   Ústrr    ÚboolrB   rK   rV   © 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!   