AX-Image-2.1-Turbo-AXQ-NVFP4 / axq-diffusion-schema.json
AutomatosX's picture
Publish research-only AXQuant NVFP4 reference storage with explicit BF16 loader
aad6de7 verified
Raw History Blame Contribute Delete
4.87 kB
{
"$defs": {
"Component": {
"additionalProperties": false,
"properties": {
"class_name": {
"enum": [
"QwenImage21Transformer2DModel",
"MiniMaxH3Transformer3DModel"
],
"title": "Class Name",
"type": "string"
},
"preserved_tensors": {
"minimum": 0,
"title": "Preserved Tensors",
"type": "integer"
},
"quantized_matrices": {
"exclusiveMinimum": 0,
"title": "Quantized Matrices",
"type": "integer"
},
"tensors": {
"additionalProperties": {
"$ref": "#/$defs/Tensor"
},
"title": "Tensors",
"type": "object"
}
},
"required": [
"class_name",
"tensors",
"quantized_matrices",
"preserved_tensors"
],
"title": "Component",
"type": "object"
},
"FileDigest": {
"additionalProperties": false,
"properties": {
"sha256": {
"pattern": "^[0-9a-f]{64}$",
"title": "Sha256",
"type": "string"
},
"size_bytes": {
"minimum": 0,
"title": "Size Bytes",
"type": "integer"
}
},
"required": [
"sha256",
"size_bytes"
],
"title": "FileDigest",
"type": "object"
},
"Tensor": {
"additionalProperties": false,
"properties": {
"dtype": {
"enum": [
"bfloat16",
"float32",
"float16",
"int64",
"int32",
"bool"
],
"title": "Dtype",
"type": "string"
},
"file": {
"title": "File",
"type": "string"
},
"global_scale": {
"anyOf": [
{
"exclusiveMinimum": 0,
"type": "number"
},
{
"type": "null"
}
],
"default": null,
"title": "Global Scale"
},
"group_size": {
"anyOf": [
{
"type": "integer"
},
{
"type": "null"
}
],
"default": null,
"title": "Group Size"
},
"method": {
"enum": [
"preserve",
"nvfp4"
],
"title": "Method",
"type": "string"
},
"shape": {
"items": {
"type": "integer"
},
"title": "Shape",
"type": "array"
}
},
"required": [
"shape",
"dtype",
"file",
"method"
],
"title": "Tensor",
"type": "object"
}
},
"additionalProperties": false,
"properties": {
"backend": {
"const": "numpy-reference",
"default": "numpy-reference",
"title": "Backend",
"type": "string"
},
"components": {
"additionalProperties": {
"$ref": "#/$defs/Component"
},
"title": "Components",
"type": "object"
},
"execution": {
"const": "reconstruct-bfloat16-diffusers",
"default": "reconstruct-bfloat16-diffusers",
"title": "Execution",
"type": "string"
},
"files": {
"additionalProperties": {
"$ref": "#/$defs/FileDigest"
},
"title": "Files",
"type": "object"
},
"format": {
"const": "nvfp4",
"default": "nvfp4",
"title": "Format",
"type": "string"
},
"native_low_precision_verified": {
"const": false,
"default": false,
"title": "Native Low Precision Verified",
"type": "boolean"
},
"quality_certified": {
"const": false,
"default": false,
"title": "Quality Certified",
"type": "boolean"
},
"schema_version": {
"const": "axq-diffusion.reference/1",
"default": "axq-diffusion.reference/1",
"title": "Schema Version",
"type": "string"
},
"source_files": {
"additionalProperties": {
"$ref": "#/$defs/FileDigest"
},
"title": "Source Files",
"type": "object"
},
"source_model": {
"enum": [
"Qwen/Qwen-Image-2.1-Turbo",
"MiniMaxAI/MiniMax-H3"
],
"title": "Source Model",
"type": "string"
},
"source_revision": {
"pattern": "^[0-9a-f]{40}$",
"title": "Source Revision",
"type": "string"
},
"status": {
"const": "development",
"default": "development",
"title": "Status",
"type": "string"
}
},
"required": [
"source_model",
"source_revision",
"source_files",
"components",
"files"
],
"title": "DiffusionManifest",
"type": "object"
}