Ë
    Gêñi½  ã                   óÊ   — 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dede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          úe/var/www/pod-logistic/pod-ai/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 ]   }t        |t        «      s|j                  «       }|j	                  d«      rX|dd  }t
        j                  d|› �«       |dk(  r y |dk(  rt
        j                  d«       Œst
        j                  d|› �«       ŒŒt
        j                  |«       Œ¢ y )Nzevent:é   zSafetensors conversion status: ÚcompleteÚ	heartbeatÚ	HeartbeatzUnknown status )Ú
iter_linesÚ
isinstanceÚstrÚdecodeÚ
startswithÚloggerÚdebug)Ú_sse_connectionÚliner   s      r   Ústartzspawn_conversion.<locals>.start   s˜   € Ø#×.Ñ.Ó0ò 	#ˆDÜ˜d¤CÔ(Ø—{‘{“}�Ø�‰˜xÔ(Ø˜a˜b˜�Ü—‘Ð>¸v¸hÐGÔHà˜ZÒ'ÙØ˜{Ò*Ü—L‘L Õ-ä—L‘L ?°6°(Ð!;Õ<ä—‘˜TÕ"ñ	#r!   ÚdataT)Úfollow_redirectsÚjsonÚevent_idÚGETú/z*Spawning safetensors automatic conversion.zError during conversion: )
r.   ÚinfoÚhttpxÚpostr5   Ústreamr/   Ú	ExceptionÚwarningÚrepr)r   r"   r   Úsafetensors_convert_space_urlÚsse_urlr2   r3   Úresultr6   Ússe_connectionÚes              r   Úspawn_conversionrE      sã   € Ü
‡K�KÐMÔNà$JÐ!Ø.Ð/¨yÐ9€Gò#ð" �X˜w¨Ð.Ð/€Dä�Z‰Z˜°$¸TÔB×GÑGÓI€FØ�jÑ!€Hä	�‰�e ˜y¨¨(¨Ð4Ó	5ð B¸ð	BÜ�L‰LÐEÔFÙ�.Ô!÷Bð Bøô ò 	BÜ�N‰NÐ6´t¸A³w°iÐ@×AÑAûð	Bú÷	Bð Bús0   Á-CÁ/BÂ	C	Â!CÂ?CÃC	Ã	CÃCc                 ól  — | j                  |«      j                  }t        j                  d«       d}|j	                  d«      }t        | |||¬«      }|�|s,|j                  dk7  rt        |||«       t        | |||¬«      }nt        j                  d«       |€t        d«      ‚d|j                  › �}|S )	Nz(Attempting to create safetensors variantz*Adding `safetensors` variant of this modelr   r   ÚSFconvertbotzSafetensors PR existszÙCould 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.   r9   Úgetr    ÚauthorrE   ÚOSErrorÚnum)r   r   Úkwargsr"   r   r   ÚprÚshas           r   Úget_conversion_pr_referencerP   ;   s²   € Ø�n‰n˜XÓ&×.Ñ.€Gä
‡K�KÐ:Ô;Ø;€HØ�J‰J�wÓ€Eô
 
�S˜( H°EÔ	:€Bà	€z™' b§i¡i°>Ò&AÜ˜ ¨Ô2Ü˜˜h¨¸Ô>‰ä�‰Ð+Ô,Ø	€zÜðgó
ð 	
ð
 �R—V‘V�HÐ
€Cà€Jr!   Úpretrained_model_name_or_pathÚignore_errors_during_conversionÚsafe_weights_nameÚsafe_weights_index_namec                 ó,  — 	 t        |j                  d«      dt        «       i¬«      }t        || fi |¤Ž}|€y||d<   |d= |j	                  | |||j                  d«      ¬«      }|r|n|}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_hashr   )r   rI   r	   rP   Úfile_existsr   r=   )rQ   rR   rS   rT   Úcached_file_kwargsr   rO   ÚshardedÚfilenameÚresolved_archive_filerD   s              r   Úauto_conversionr]   W   sÍ   € ðÜÐ,×0Ñ0°Ó9ÀLÔRaÓRcÐCdÔeˆÜ)¨#Ð/LÑcÐPbÑcˆàˆ;ØØ),Ð˜:Ñ&Ø˜~Ð.ð —/‘/Ø)Ø#ØØ$×(Ñ(¨Ó1ð	 "ó 
ˆñ /6Ñ*Ð;Lˆä +Ð,IÈ8Ñ jÐWiÑ jÐØ$ c¨7Ð2Ð2øÜò Ù.ØˆGô /ûðús   ‚5A< ¸AA< Á<	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    ÚboolrE   rP   r]   © r!   r   ú<module>re      s¿   ðÝ ã ß CÑ Cç 8Ñ 8ð 
ˆ×	Ñ	˜HÓ	%€ð�Uð  cð °Sð Àð ÈÐR^ÑI_ó ð!B˜Cð !B¨$ð !B¸#ó !BðH Uð °có ð< -2Ø0Ø#Añ	Ø#&ðà%)ðð ðð !ô	r!   