L i ddlmZddlZddlmZmZmZddlmZm Z m Z e je Z dededed ed ed f d Zd ed edefdZdedefdZddefdZy))OptionalN) DiscussionHfApiget_repo_discussions) cached_filehttp_user_agentloggingapimodel_idpr_titletokenreturnrc*|j||dj}t||D]c}|j|k(s|jdk(s#|j s0|j||j |}||djk(sa|cSy)Nrr)repo_idropenrevisionrr)list_repo_commits commit_idrtitlestatusis_pull_request git_reference)r r r r main_commit discussioncommitss i/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/transformers/safetensors_conversion.py previous_prr s'''>qAKKK*85I"   x 'J,=,=,GJLfLf++Hz?W?W_d+eGgaj222!! " privatectjdd}|d}d}d|||gi}tj|d|j }|d}tj |d |d 5} tj d || dddy#t$r+} tjd t| Yd} ~ 8d} ~ wwxYw#1swYyxYw) Nz;Attempting to convert .bin model on the fly to safetensors.z$https://safetensors-convert.hf.spacez /call/runcJ|jD]}|j}|jdrX|dd}tj d||dk(ry|dk(rtj dctj d||tj |y)Nzevent:zSafetensors conversion status: complete heartbeat HeartbeatzUnknown status ) iter_linesdecode startswithloggerdebug)_sse_connectionliners rstartzspawn_conversion..starts#..0 #D;;=Dx(ab >vhGHZ'{*LL-LL?6(!;< T" #r!dataT)streamjsonevent_id/)r2z*Spawning safetensors automatic conversion.zError during conversion: ) r,inforequestspostr3getr- Exceptionwarningrepr) rr"r safetensors_convert_space_urlsse_urlr0r1resultr4sse_connectiones rspawn_conversionrBs KKMN$J!./y9G# Xw. /D ]]74d ; @ @ BFj!H  8*-d ;B~ B LLE F . !BB B NN6tAwi@ A A B BBs0.C 0B C !CC C  C  Cc R|j|j}tjdd}|j d}t ||||}||s,|j dk7rt|||t ||||}ntjdd|j}|S)Nz(Attempting to create safetensors variantz*Adding `safetensors` variant of this modelrr SFconvertbotzSafetensors PR existszrefs/pr/) model_infor"r,r6r9r authorrBnum)r r kwargsr"r rprshas rget_conversion_pr_referencerK:snnX&..G KK:;;H JJw E S(HE :B z'bii>&A2 h > +, RVVH C Jr!pretrained_model_name_or_pathc , t|jddti}t||fi|}|y||d<|d=|j |d||jd}|rdnd }t ||fi|}|||fS#t $r}|s|Yd}~yd}~wwxYw) Nrz user-agent)rheaders)NNr _commit_hashzmodel.safetensors.index.jsonrzmodel.safetensors)rr9r rK file_existsrr:) rLignore_errors_during_conversioncached_file_kwargsr rJshardedfilenameresolved_archive_filerAs rauto_conversionrVQs,009LRaRcCde)#/LcPbc ;),:& ~ .// ) *$((1 "  6=1BU +,I8 jWi j$c722 .G/s5A<AA<< BBB)F)typingrr7huggingface_hubrrrutilsrr r get_logger__name__r,strr boolrBrKrVr!rr_sCC88   H %UcSR^I_ BC B$ B# BFUc.3r!