
    Y jJ                         S SK r S SKrS SKJr  S SKJrJr  S SKJr  \	r
\	r\S   r\ " S S5      5       r\ " S S	5      5       r\ " S
 S5      5       rg)    N)defaultdict)	dataclassfield)Literal)
F64F32F16BF16I64I32I16I8U8BOOLc                   h    \ rS rSr% Sr\\S'   \\   \S'   \	\\4   \S'   \
" SS9r\\S'   SS
 jrSrg	)
TensorInfo   a  Information about a tensor.

For more details regarding the safetensors format, check out https://huggingface.co/docs/safetensors/index#format.

Attributes:
    dtype (`str`):
        The data type of the tensor ("F64", "F32", "F16", "BF16", "I64", "I32", "I16", "I8", "U8", "BOOL").
    shape (`list[int]`):
        The shape of the tensor.
    data_offsets (`tuple[int, int]`):
        The offsets of the data in the file as a tuple `[BEGIN, END]`.
    parameter_count (`int`):
        The number of parameters in the tensor.
dtypeshapedata_offsetsFinitparameter_countNc                      [         R                  " [        R                  U R                  5      U l        g ! [         a
    SU l         g f = f)N   )	functoolsreduceoperatormulr   r   	TypeError)selfs    s/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/huggingface_hub/utils/_safetensors.py__post_init__TensorInfo.__post_init__#   s;    	%#,#3#3HLL$**#MD  	%#$D 	%s   47 A
Ar   returnN)__name__
__module____qualname____firstlineno____doc__DTYPE_T__annotations__listinttupler   r   r#   __static_attributes__     r"   r   r      s:     N9S/! e,OS,%r4   r   c                   l    \ rS rSr% Sr\\\4   \S'   \\\	4   \S'   \
" SS9r\\\4   \S'   SS	 jrS
rg)SafetensorsFileMetadata+   a  Metadata for a Safetensors file hosted on the Hub.

This class is returned by [`parse_safetensors_file_metadata`].

For more details regarding the safetensors format, check out https://huggingface.co/docs/safetensors/index#format.

Attributes:
    metadata (`dict`):
        The metadata contained in the file.
    tensors (`dict[str, TensorInfo]`):
        A map of all tensors. Keys are tensor names and values are information about the corresponding tensor, as a
        [`TensorInfo`] object.
    parameter_count (`dict[str, int]`):
        A map of the number of parameters per data type. Keys are data types and values are the number of parameters
        of that data type.
metadatatensorsFr   r   Nc                     [        [        5      nU R                  R                  5        H#  nXR                  ==   UR
                  -  ss'   M%     [        U5      U l        g N)r   r0   r9   valuesr   r   dict)r!   r   tensors      r"   r#   %SafetensorsFileMetadata.__post_init__B   sH    .9#.>ll))+FLL)V-C-CC) ,#O4r4   r%   r&   )r(   r)   r*   r+   r,   r=   strr.   TENSOR_NAME_Tr   r   r   r-   r0   r#   r2   r3   r4   r"   r6   r6   +   sD    " 38n-+,,*/U*;OT'3,';5r4   r6   c                       \ rS rSr% Sr\S-  \S'   \\S'   \\\	4   \S'   \\	\
4   \S'   \" SS	9r\\\4   \S
'   SS jrSrg)SafetensorsRepoMetadataI   a  Metadata for a Safetensors repo.

A repo is considered to be a Safetensors repo if it contains either a 'model.safetensors' weight file (non-shared
model) or a 'model.safetensors.index.json' index file (sharded model) at its root.

This class is returned by [`get_safetensors_metadata`].

For more details regarding the safetensors format, check out https://huggingface.co/docs/safetensors/index#format.

Attributes:
    metadata (`dict`, *optional*):
        The metadata contained in the 'model.safetensors.index.json' file, if it exists. Only populated for sharded
        models.
    sharded (`bool`):
        Whether the repo contains a sharded model or not.
    weight_map (`dict[str, str]`):
        A map of all weights. Keys are tensor names and values are filenames of the files containing the tensors.
    files_metadata (`dict[str, SafetensorsFileMetadata]`):
        A map of all files metadata. Keys are filenames and values are the metadata of the corresponding file, as
        a [`SafetensorsFileMetadata`] object.
    parameter_count (`dict[str, int]`):
        A map of the number of parameters per data type. Keys are data types and values are the number of parameters
        of that data type.
Nr8   sharded
weight_mapfiles_metadataFr   r   c                     [        [        5      nU R                  R                  5        H2  nUR                  R                  5        H  u  p4X==   U-  ss'   M     M4     [        U5      U l        g r;   )r   r0   rG   r<   r   itemsr=   )r!   r   file_metadatar   nb_parameters_s        r"   r#   %SafetensorsRepoMetadata.__post_init__j   s_    .9#.>!00779M)6)F)F)L)L)N%&.8& *O :  $O4r4   r%   r&   )r(   r)   r*   r+   r,   r=   r.   boolrA   
FILENAME_Tr6   r   r   r-   r0   r#   r2   r3   r4   r"   rC   rC   I   sV    2 TkM]J.//%<<==*/U*;OT'3,';5r4   rC   )r   r   collectionsr   dataclassesr   r   typingr   r@   rN   rA   r-   r   r6   rC   r3   r4   r"   <module>rR      sv      # (  

V
W % % %: 5 5 5: %5 %5 %5r4   