Python module

max.graph.weights

Weight containers

WeightData	Container for weight tensor data with metadata.
Weights	Protocol for managing and accessing model weights hierarchically.
WeightsAdapter	alias of Callable[[...], dict[str, WeightData]]

Format-specific loaders

GGUFWeights	Implementation for loading weights from GGUF (GPT-Generated Unified Format) files.
SafetensorWeights	Implementation for loading weights from safetensors files.

Loading functions

load_weights	Loads neural network weights from checkpoint files.
weights_format	Detect the format of weight files based on their extensions.

Enums

WeightsFormat	Enumeration of supported weight file formats.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit