vllm_omni.host_weight_runtime.filesystem.writer ¶

Store-scoped streaming safetensors publication.

ArtifactWriterError ¶

Scoped writer that cannot publish outside one invisible staging dir.

staging_dir = staging_dir

abort() -> None

finalize(
    identity: WeightArtifactIdentity,
    metadata: ProductionMetadata,
) -> tuple[WeightManifest, PublicationMarker]

open_tensor_file(
    relative_name: str,
    specs: tuple[TensorWriteSpec, ...],
    *,
    ordered: bool = False,
) -> StreamingSafeTensorFile

One preallocated safetensors file filled through bounded CPU writes.

relative_name = _safe_relative_name(relative_name)

specs = tuple(sorted(specs, key=lambda spec: spec.name))

abort() -> None

close() -> None

write_rows(
    name: str, start_row: int, tensor: Tensor
) -> None

write_tensor(name: str, tensor: Tensor) -> None