Text-to-Image
Diffusers
Safetensors
AXQReferenceDiffusionPipeline
axquant
axq
nvfp4
reference-storage
research-only
development
Instructions to use AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Download axq-diffusion-schema.json from AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4: direct link, hf CLI and curl.
- Browser
- Download file 4.87 kB
-
https://huggingface.co/AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4/resolve/main/axq-diffusion-schema.json
- Command line
-
hf download hf://AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4/axq-diffusion-schema.json
-
curl -L -o axq-diffusion-schema.json https://huggingface.co/AutomatosX/AX-Image-2.1-Turbo-AXQ-NVFP4/resolve/main/axq-diffusion-schema.json
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" | |
| } | |