torch_io.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.torch_io

Functions
bool	aquin.compute.torch_io.is_corrupt_pytorch_zip (BaseException exc)
Any	aquin.compute.torch_io.load_checkpoint (str|Path path, *, Any map_location=None)
dict[str, Any]|None	aquin.compute.torch_io.load_norm_stats (str|Path path, *, Any map_location=None)
bool	aquin.compute.torch_io.looks_like_catalog_metadata_blob (str|Path path)
bool	aquin.compute.torch_io.looks_like_invalid_norm_blob (str|Path path)
bool	aquin.compute.torch_io.looks_like_json_error_blob (str|Path path)
bool	aquin.compute.torch_io.looks_like_catalog_metadata_dict (Any data)
dict[str, Any]	aquin.compute.torch_io._checkpoint_tensors (Any ckpt)
bool	aquin.compute.torch_io._has_sae_weight_tensors (dict[str, Any] tensors)
bool	aquin.compute.torch_io.is_valid_sae_checkpoint_path (str|Path path)