module ptwm.integrations

def compress_safetensors_file(src_path: str | Path, out_dir: str | Path, mode: Literal['a', 'b'] = 'a', classifier: TensorClassifier | ClassifierChain | None = None, max_shard_size: int = _DEFAULT_MAX_SHARD_SIZE, method_hint: int = 3, explore_options: ExploreOptions | None = None, use_user_cache: bool = True, fast: bool = False, fast_chunk: int = 512 * 1024) -> AuditLog

def discover_ptwm_shards(model_dir: str | os.PathLike[str]) -> list[Path]

def patch_safetensors() -> None

def patch_transformers() -> None

class SafeOpen:

def __enter__(self) -> Self

def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None

def __init__(self, filename: str, framework: str, device: str = 'cpu') -> None

compressed_tensors_metadata: dict[str, Any] = {}

def get_slice(self, name: str) -> Any

def get_tensor(self, name: str) -> torch.Tensor

def keys(self) -> list[str]

def metadata(self) -> dict[str, str] | None