Image-Text-to-Text
Safetensors
MLX
mlx-vlm
mistral3
apple-silicon
pixtral
guardrail
content-moderation
safety-classification
multimodal
4-bit precision
conversational
Instructions to use AXONVERTEX-AI-RESEARCH/Shieldstral-1.0-3B-MLX-4bit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use AXONVERTEX-AI-RESEARCH/Shieldstral-1.0-3B-MLX-4bit with MLX:
# Make sure mlx-vlm is installed # pip install --upgrade mlx-vlm from mlx_vlm import load, generate from mlx_vlm.prompt_utils import apply_chat_template from mlx_vlm.utils import load_config # Load the model model, processor = load("AXONVERTEX-AI-RESEARCH/Shieldstral-1.0-3B-MLX-4bit") config = load_config("AXONVERTEX-AI-RESEARCH/Shieldstral-1.0-3B-MLX-4bit") # Prepare input image = ["http://images.cocodataset.org/val2017/000000039769.jpg"] prompt = "Describe this image." # Apply chat template formatted_prompt = apply_chat_template( processor, config, prompt, num_images=1 ) # Generate output output = generate(model, processor, formatted_prompt, image) print(output) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
| { | |
| "dim": 3072, | |
| "n_layers": 26, | |
| "head_dim": 128, | |
| "hidden_dim": 9216, | |
| "n_heads": 32, | |
| "n_kv_heads": 8, | |
| "rope_theta": 1000000.0, | |
| "norm_eps": 1e-05, | |
| "vocab_size": 131072, | |
| "tied_embeddings": true, | |
| "max_position_embeddings": 262144, | |
| "ragged_attention": null, | |
| "nope": null, | |
| "softmax_tanh_gating": null, | |
| "llama_4_scaling": { | |
| "original_max_position_embeddings": 16384, | |
| "beta": 0.1 | |
| }, | |
| "q_lora_rank": null, | |
| "qk_rope_head_dim": null, | |
| "qk_nope_head_dim": null, | |
| "kv_lora_rank": null, | |
| "v_head_dim": null, | |
| "yarn": { | |
| "original_max_position_embeddings": 16384, | |
| "factor": 16, | |
| "apply_scale": false, | |
| "beta": 32, | |
| "alpha": 1 | |
| }, | |
| "kv_sharing": null, | |
| "moe": null, | |
| "vision_encoder": { | |
| "image_token_id": 10, | |
| "image_break_token_id": 12, | |
| "image_end_token_id": 13, | |
| "intermediate_size": 4096, | |
| "num_hidden_layers": 24, | |
| "num_attention_heads": 16, | |
| "mm_projector_id": "patch_merge", | |
| "spatial_merge_size": 2, | |
| "hidden_size": 1024, | |
| "num_channels": 3, | |
| "image_size": 1540, | |
| "max_image_size": 1540, | |
| "patch_size": 14, | |
| "rope_theta": 10000.0, | |
| "add_pre_mm_projector_layer_norm": true, | |
| "adapter_bias": false | |
| } | |
| } |