o
    i                     @   sL   d Z ddlmZmZ ddlmZmZ ddlmZm	Z	m
Z
mZmZmZmZ dS )z&Contains helpers to serialize tensors.   )StateDictSplit$split_state_dict_into_shards_factory)get_tf_storage_sizesplit_tf_state_dict_into_shards)get_torch_storage_idget_torch_storage_sizeload_state_dict_from_fileload_torch_modelsave_torch_modelsave_torch_state_dict"split_torch_state_dict_into_shardsN)__doc___baser   r   _tensorflowr   r   _torchr   r   r   r	   r
   r   r    r   r   c/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/huggingface_hub/serialization/__init__.py<module>   s   (