o ‡ZŽ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)%ÚrZ _hf_storagerrÚapirZdefault_plannerrrÚ filesystemrr Úmetadatar r r r Z optimizerrZplannerrrrrrrZstate_dict_loaderrrZstate_dict_saverrrrZstoragerr©r r úT/var/www/auris/lib/python3.10/site-packages/torch/distributed/checkpoint/__init__.pyÚs