Ë
     õpiú  ã                   óÐ   — d dl mZ d dlmZ d dlmZm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 d dlmZ d dlmZmZmZmZmZmZ d d	lmZ d d
lmZmZ d dl m!Z!m"Z"m#Z# d dl$m%Z%m&Z& y)é   )Ú
_extension)ÚCheckpointException)ÚDefaultLoadPlannerÚDefaultSavePlanner)ÚFileSystemReaderÚFileSystemWriter)ÚHuggingFaceStorageReaderÚHuggingFaceStorageWriter)ÚBytesStorageMetadataÚChunkStorageMetadataÚMetadataÚTensorStorageMetadata)Ú!load_sharded_optimizer_state_dict)ÚLoadPlanÚLoadPlannerÚReadItemÚSavePlanÚSavePlannerÚ	WriteItem)Ú!QuantizedHuggingFaceStorageReader)ÚloadÚload_state_dict)Ú
async_saveÚsaveÚsave_state_dict)ÚStorageReaderÚStorageWriterN)'Ú r   Úapir   Údefault_plannerr   r   Ú
filesystemr   r   Ú
hf_storager	   r
   Úmetadatar   r   r   r   Ú	optimizerr   Úplannerr   r   r   r   r   r   Úquantized_hf_storager   Ústate_dict_loaderr   r   Ústate_dict_saverr   r   r   Ústorager   r   © ó    úg/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torch/distributed/checkpoint/__init__.pyú<module>r-      s;   ðÝ Ý $ß Cß :ß J÷ó õ 9ß V× VÝ Cß 4ß ?Ñ ?ß 1Ð 1r+   