o Hñ h¹ã@sÄddlmZddlmZmZddlmZddlmZm Z ddl m Z m Z ddl mZmZmZmZddlmZddlmZmZmZmZmZmZdd lmZmZdd lmZm Z m!Z!dd l"m#Z#m$Z$d S) é)Ú _extension)Ú_HuggingFaceStorageReaderÚ_HuggingFaceStorageWriter)ÚCheckpointException)ÚDefaultLoadPlannerÚDefaultSavePlanner)ÚFileSystemReaderÚFileSystemWriter)ÚBytesStorageMetadataÚChunkStorageMetadataÚMetadataÚTensorStorageMetadata)Ú!load_sharded_optimizer_state_dict)ÚLoadPlanÚ LoadPlannerÚReadItemÚSavePlanÚ SavePlannerÚ WriteItem)ÚloadÚload_state_dict)Ú async_saveÚsaveÚsave_state_dict)Ú StorageReaderÚ StorageWriterN)%ÚrÚ _hf_storagerrÚapirÚdefault_plannerrrÚ filesystemrr Úmetadatar r r r Ú optimizerrÚplannerrrrrrrÚstate_dict_loaderrrÚstate_dict_saverrrrÚstoragerr©r'r'úY/var/www/vscode/kcb/lib/python3.10/site-packages/torch/distributed/checkpoint/__init__.pyÚs