docs/finetrainers-src-codebase/finetrainers/utils/serialization.py

from typing import Any, Dict, Optional

import safetensors.torch


def safetensors_torch_save_function(weights: Dict[str, Any], filename: str, metadata: Optional[Dict[str, str]] = None):
    if metadata is None:
        metadata = {}
    metadata["format"] = "pt"
    safetensors.torch.save_file(weights, filename, metadata)