o հi1@sddlZddlZddlZddlZddlmZdejvr5eejjjjd r5ej de eejjjddl mZedZde ddfd d Zd ed ejddfd dZdddZedkrcedSdS)N)Path NO_LOCAL_GGUFzgguf-py) GGUFReaderzgguf-set-metadatafilenamereturncCs<t|d}|jd}|durdS|jd}d|j|d<dS)Nr+ztokenizer.ggml.bos_token_idr)rfieldsdataparts)rreaderfield part_indexrR/home/ubuntu/.local/lib/python3.10/site-packages/gguf/scripts/gguf_set_metadata.pyminimal_examples   rr argscCsj||j}|durtdt|jdtd|jr'|j |jdnd}|durAtdt|jd|jtd|j |j dd}||j }t dt|jd|d |||krxt d t|jd |td|jrtd|jstd td tdtd}|dkrt dtd||j |j dd<t ddS)Nz! Field z not foundrz2! This tool only supports changing simple values, z has unsupported type z* Preparing to change field z from z to z- Key z already set to requested value z&*** Warning *** Warning *** Warning **zP* Changing fields in a GGUF file can make it unusable. Proceed at your own risk.z<* Enter exactly YES if you are positive you want to proceed:zYES, I am sure> YESz(You didn't enter YES. Okay then, see ya!z'* Field changed. Successful completion.) get_fieldkeyloggererrorreprsysexittypesgguf_scalar_to_npgetr r valueinfodry_runforcewarninginput)r rr handler current_value new_valueresponserrr set_metadata,s4    "       r)cCstjdd}|jdtdd|jdtdd|jdtd d|jd d d d |jdd dd |jdd dd |ttjdkrAdndg}tj |j rMtj ntj dt d|jt|j|jrcdnd}t||dS)Nz(Set a simple value in GGUF file metadata) descriptionmodelzGGUF format model filename)typehelprzMetadata key to setrzMetadata value to setz --dry-run store_truezDon't actually change anything)actionr-z--forcez%Change the field without confirmationz --verbosezincrease output verbosityrz--help)levelz * Loading: rr)argparseArgumentParser add_argumentstr parse_argslenrargvlogging basicConfigverboseDEBUGINFOrr r+rr!r))parserrr rrrmainLs r?__main__)rN)r9r2osrpathlibrenviron__file__parentexistspathinsertr5ggufr getLoggerrr Namespacer)r?__name__rrrrs "