Instructions to use tiny-random/inkling-2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use tiny-random/inkling-2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-text-to-text", model="tiny-random/inkling-2", trust_remote_code=True) messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] pipe(text=messages)# pip install -U transformers accelerate # Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("tiny-random/inkling-2", trust_remote_code=True) model = AutoModelForMultimodalLM.from_pretrained("tiny-random/inkling-2", trust_remote_code=True, device_map="auto") messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] inputs = processor.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=256) print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use tiny-random/inkling-2 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "tiny-random/inkling-2" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "tiny-random/inkling-2", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker
docker model run hf.co/tiny-random/inkling-2
- SGLang
How to use tiny-random/inkling-2 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "tiny-random/inkling-2" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "tiny-random/inkling-2", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "tiny-random/inkling-2" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "tiny-random/inkling-2", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }' - Docker Model Runner
How to use tiny-random/inkling-2 with Docker Model Runner:
docker model run hf.co/tiny-random/inkling-2
Download hf_quant_config.json from tiny-random/inkling-2: direct link, hf CLI and curl.
- Browser
- Download file 24.9 kB
-
https://huggingface.co/tiny-random/inkling-2/resolve/main/hf_quant_config.json
- Command line
-
hf download hf://tiny-random/inkling-2/hf_quant_config.json
-
curl -L -o hf_quant_config.json https://huggingface.co/tiny-random/inkling-2/resolve/main/hf_quant_config.json
24.9 kB
| { | |
| "model": { | |
| "parameter_groups": [ | |
| { | |
| "compute_dtype": "float32", | |
| "output_dtype": "float32", | |
| "required_activation_quantization": null, | |
| "tensors": [ | |
| "model.llm.attn_res.attn_queries.0", | |
| "model.llm.attn_res.attn_queries.1", | |
| "model.llm.attn_res.attn_queries.10", | |
| "model.llm.attn_res.attn_queries.2", | |
| "model.llm.attn_res.attn_queries.3", | |
| "model.llm.attn_res.attn_queries.4", | |
| "model.llm.attn_res.attn_queries.5", | |
| "model.llm.attn_res.attn_queries.6", | |
| "model.llm.attn_res.attn_queries.7", | |
| "model.llm.attn_res.attn_queries.8", | |
| "model.llm.attn_res.attn_queries.9", | |
| "model.llm.attn_res.mlp_queries.0", | |
| "model.llm.attn_res.mlp_queries.1", | |
| "model.llm.attn_res.mlp_queries.10", | |
| "model.llm.attn_res.mlp_queries.11", | |
| "model.llm.attn_res.mlp_queries.2", | |
| "model.llm.attn_res.mlp_queries.3", | |
| "model.llm.attn_res.mlp_queries.4", | |
| "model.llm.attn_res.mlp_queries.5", | |
| "model.llm.attn_res.mlp_queries.6", | |
| "model.llm.attn_res.mlp_queries.7", | |
| "model.llm.attn_res.mlp_queries.8", | |
| "model.llm.attn_res.mlp_queries.9", | |
| "model.llm.attn_res.output_query", | |
| "model.llm.layers.0.mlp.gate.bias", | |
| "model.llm.layers.1.mlp.gate.bias", | |
| "model.llm.layers.10.mlp.gate.bias", | |
| "model.llm.layers.11.mlp.gate.bias", | |
| "model.llm.layers.2.mlp.gate.bias", | |
| "model.llm.layers.3.mlp.gate.bias", | |
| "model.llm.layers.4.mlp.gate.bias", | |
| "model.llm.layers.5.mlp.gate.bias", | |
| "model.llm.layers.6.mlp.gate.bias", | |
| "model.llm.layers.7.mlp.gate.bias", | |
| "model.llm.layers.8.mlp.gate.bias", | |
| "model.llm.layers.9.mlp.gate.bias" | |
| ], | |
| "weight_block_shape": [], | |
| "weight_transform": "cast" | |
| }, | |
| { | |
| "compute_dtype": "bfloat16", | |
| "output_dtype": "bfloat16", | |
| "required_activation_quantization": null, | |
| "tensors": [ | |
| "model.llm.embed.weight", | |
| "model.llm.embed_norm.weight", | |
| "model.llm.layers.0.attn.attn_sink_proj.weight", | |
| "model.llm.layers.0.attn.gate_attn.weight", | |
| "model.llm.layers.0.attn.k_sconv.weight", | |
| "model.llm.layers.0.attn.q_l2_clip.scale", | |
| "model.llm.layers.0.attn.rel_l2_clip.scale", | |
| "model.llm.layers.0.attn.rel_logits_proj.proj", | |
| "model.llm.layers.0.attn.v_sconv.weight", | |
| "model.llm.layers.0.attn.wk_dv.weight", | |
| "model.llm.layers.0.attn.wo_ud.weight", | |
| "model.llm.layers.0.attn.wq_du.weight", | |
| "model.llm.layers.0.attn.wr_du.weight", | |
| "model.llm.layers.0.attn.wv_dv.weight", | |
| "model.llm.layers.0.attn_norm.weight", | |
| "model.llm.layers.0.attn_sconv.weight", | |
| "model.llm.layers.0.mlp.gate.weight", | |
| "model.llm.layers.0.mlp_norm.weight", | |
| "model.llm.layers.0.mlp_sconv.weight", | |
| "model.llm.layers.1.attn.attn_sink_proj.weight", | |
| "model.llm.layers.1.attn.gate_attn.weight", | |
| "model.llm.layers.1.attn.k_sconv.weight", | |
| "model.llm.layers.1.attn.q_l2_clip.scale", | |
| "model.llm.layers.1.attn.rel_l2_clip.scale", | |
| "model.llm.layers.1.attn.rel_logits_proj.proj", | |
| "model.llm.layers.1.attn.v_sconv.weight", | |
| "model.llm.layers.1.attn.wk_dv.weight", | |
| "model.llm.layers.1.attn.wo_ud.weight", | |
| "model.llm.layers.1.attn.wq_du.weight", | |
| "model.llm.layers.1.attn.wr_du.weight", | |
| "model.llm.layers.1.attn.wv_dv.weight", | |
| "model.llm.layers.1.attn_norm.weight", | |
| "model.llm.layers.1.attn_sconv.weight", | |
| "model.llm.layers.1.mlp.gate.weight", | |
| "model.llm.layers.1.mlp_norm.weight", | |
| "model.llm.layers.1.mlp_sconv.weight", | |
| "model.llm.layers.10.attn.attn_sink_proj.weight", | |
| "model.llm.layers.10.attn.gate_attn.weight", | |
| "model.llm.layers.10.attn.k_sconv.weight", | |
| "model.llm.layers.10.attn.q_l2_clip.scale", | |
| "model.llm.layers.10.attn.rel_l2_clip.scale", | |
| "model.llm.layers.10.attn.rel_logits_proj.proj", | |
| "model.llm.layers.10.attn.v_sconv.weight", | |
| "model.llm.layers.10.attn.wk_dv.weight", | |
| "model.llm.layers.10.attn.wo_ud.weight", | |
| "model.llm.layers.10.attn.wq_du.weight", | |
| "model.llm.layers.10.attn.wr_du.weight", | |
| "model.llm.layers.10.attn.wv_dv.weight", | |
| "model.llm.layers.10.attn_norm.weight", | |
| "model.llm.layers.10.attn_sconv.weight", | |
| "model.llm.layers.10.mlp.gate.weight", | |
| "model.llm.layers.10.mlp_norm.weight", | |
| "model.llm.layers.10.mlp_sconv.weight", | |
| "model.llm.layers.11.attn.attn_sink_proj.weight", | |
| "model.llm.layers.11.attn.gate_attn.weight", | |
| "model.llm.layers.11.attn.k_sconv.static_w", | |
| "model.llm.layers.11.attn.k_sconv.weight_proj", | |
| "model.llm.layers.11.attn.q_l2_clip.scale", | |
| "model.llm.layers.11.attn.rel_l2_clip.scale", | |
| "model.llm.layers.11.attn.rel_logits_proj.proj", | |
| "model.llm.layers.11.attn.v_sconv.static_w", | |
| "model.llm.layers.11.attn.v_sconv.weight_proj", | |
| "model.llm.layers.11.attn.wk_dv.weight", | |
| "model.llm.layers.11.attn.wo_ud.weight", | |
| "model.llm.layers.11.attn.wq_du.weight", | |
| "model.llm.layers.11.attn.wr_du.weight", | |
| "model.llm.layers.11.attn.wv_dv.weight", | |
| "model.llm.layers.11.attn_sconv.weight", | |
| "model.llm.layers.11.k_input_norm.weight", | |
| "model.llm.layers.11.mlp.gate.weight", | |
| "model.llm.layers.11.mlp_norm.weight", | |
| "model.llm.layers.11.mlp_sconv.weight", | |
| "model.llm.layers.11.q_input_norm.weight", | |
| "model.llm.layers.11.v_input_norm.weight", | |
| "model.llm.layers.2.attn.attn_sink_proj.weight", | |
| "model.llm.layers.2.attn.gate_attn.weight", | |
| "model.llm.layers.2.attn.k_sconv.weight", | |
| "model.llm.layers.2.attn.q_l2_clip.scale", | |
| "model.llm.layers.2.attn.rel_l2_clip.scale", | |
| "model.llm.layers.2.attn.rel_logits_proj.proj", | |
| "model.llm.layers.2.attn.v_sconv.weight", | |
| "model.llm.layers.2.attn.wk_dv.weight", | |
| "model.llm.layers.2.attn.wo_ud.weight", | |
| "model.llm.layers.2.attn.wq_du.weight", | |
| "model.llm.layers.2.attn.wr_du.weight", | |
| "model.llm.layers.2.attn.wv_dv.weight", | |
| "model.llm.layers.2.attn_norm.weight", | |
| "model.llm.layers.2.attn_sconv.weight", | |
| "model.llm.layers.2.mlp.gate.weight", | |
| "model.llm.layers.2.mlp_norm.weight", | |
| "model.llm.layers.2.mlp_sconv.weight", | |
| "model.llm.layers.3.attn.attn_sink_proj.weight", | |
| "model.llm.layers.3.attn.gate_attn.weight", | |
| "model.llm.layers.3.attn.k_sconv.weight", | |
| "model.llm.layers.3.attn.q_l2_clip.scale", | |
| "model.llm.layers.3.attn.rel_l2_clip.scale", | |
| "model.llm.layers.3.attn.rel_logits_proj.proj", | |
| "model.llm.layers.3.attn.v_sconv.weight", | |
| "model.llm.layers.3.attn.wk_dv.weight", | |
| "model.llm.layers.3.attn.wo_ud.weight", | |
| "model.llm.layers.3.attn.wq_du.weight", | |
| "model.llm.layers.3.attn.wr_du.weight", | |
| "model.llm.layers.3.attn.wv_dv.weight", | |
| "model.llm.layers.3.attn_norm.weight", | |
| "model.llm.layers.3.attn_sconv.weight", | |
| "model.llm.layers.3.mlp.gate.weight", | |
| "model.llm.layers.3.mlp_norm.weight", | |
| "model.llm.layers.3.mlp_sconv.weight", | |
| "model.llm.layers.4.attn.attn_sink_proj.weight", | |
| "model.llm.layers.4.attn.gate_attn.weight", | |
| "model.llm.layers.4.attn.k_sconv.weight", | |
| "model.llm.layers.4.attn.q_l2_clip.scale", | |
| "model.llm.layers.4.attn.rel_l2_clip.scale", | |
| "model.llm.layers.4.attn.rel_logits_proj.proj", | |
| "model.llm.layers.4.attn.v_sconv.weight", | |
| "model.llm.layers.4.attn.wk_dv.weight", | |
| "model.llm.layers.4.attn.wo_ud.weight", | |
| "model.llm.layers.4.attn.wq_du.weight", | |
| "model.llm.layers.4.attn.wr_du.weight", | |
| "model.llm.layers.4.attn.wv_dv.weight", | |
| "model.llm.layers.4.attn_norm.weight", | |
| "model.llm.layers.4.attn_sconv.weight", | |
| "model.llm.layers.4.mlp.gate.weight", | |
| "model.llm.layers.4.mlp_norm.weight", | |
| "model.llm.layers.4.mlp_sconv.weight", | |
| "model.llm.layers.5.attn.attn_sink_proj.weight", | |
| "model.llm.layers.5.attn.gate_attn.weight", | |
| "model.llm.layers.5.attn.k_sconv.static_w", | |
| "model.llm.layers.5.attn.k_sconv.weight_proj", | |
| "model.llm.layers.5.attn.q_l2_clip.scale", | |
| "model.llm.layers.5.attn.rel_l2_clip.scale", | |
| "model.llm.layers.5.attn.rel_logits_proj.proj", | |
| "model.llm.layers.5.attn.v_sconv.static_w", | |
| "model.llm.layers.5.attn.v_sconv.weight_proj", | |
| "model.llm.layers.5.attn.wk_dv.weight", | |
| "model.llm.layers.5.attn.wo_ud.weight", | |
| "model.llm.layers.5.attn.wq_du.weight", | |
| "model.llm.layers.5.attn.wr_du.weight", | |
| "model.llm.layers.5.attn.wv_dv.weight", | |
| "model.llm.layers.5.attn_sconv.weight", | |
| "model.llm.layers.5.k_input_norm.weight", | |
| "model.llm.layers.5.mlp.gate.weight", | |
| "model.llm.layers.5.mlp_norm.weight", | |
| "model.llm.layers.5.mlp_sconv.weight", | |
| "model.llm.layers.5.q_input_norm.weight", | |
| "model.llm.layers.5.v_input_norm.weight", | |
| "model.llm.layers.6.attn.attn_sink_proj.weight", | |
| "model.llm.layers.6.attn.gate_attn.weight", | |
| "model.llm.layers.6.attn.k_sconv.weight", | |
| "model.llm.layers.6.attn.q_l2_clip.scale", | |
| "model.llm.layers.6.attn.rel_l2_clip.scale", | |
| "model.llm.layers.6.attn.rel_logits_proj.proj", | |
| "model.llm.layers.6.attn.v_sconv.weight", | |
| "model.llm.layers.6.attn.wk_dv.weight", | |
| "model.llm.layers.6.attn.wo_ud.weight", | |
| "model.llm.layers.6.attn.wq_du.weight", | |
| "model.llm.layers.6.attn.wr_du.weight", | |
| "model.llm.layers.6.attn.wv_dv.weight", | |
| "model.llm.layers.6.attn_norm.weight", | |
| "model.llm.layers.6.attn_sconv.weight", | |
| "model.llm.layers.6.mlp.gate.weight", | |
| "model.llm.layers.6.mlp_norm.weight", | |
| "model.llm.layers.6.mlp_sconv.weight", | |
| "model.llm.layers.7.attn.attn_sink_proj.weight", | |
| "model.llm.layers.7.attn.gate_attn.weight", | |
| "model.llm.layers.7.attn.k_sconv.weight", | |
| "model.llm.layers.7.attn.q_l2_clip.scale", | |
| "model.llm.layers.7.attn.rel_l2_clip.scale", | |
| "model.llm.layers.7.attn.rel_logits_proj.proj", | |
| "model.llm.layers.7.attn.v_sconv.weight", | |
| "model.llm.layers.7.attn.wk_dv.weight", | |
| "model.llm.layers.7.attn.wo_ud.weight", | |
| "model.llm.layers.7.attn.wq_du.weight", | |
| "model.llm.layers.7.attn.wr_du.weight", | |
| "model.llm.layers.7.attn.wv_dv.weight", | |
| "model.llm.layers.7.attn_norm.weight", | |
| "model.llm.layers.7.attn_sconv.weight", | |
| "model.llm.layers.7.mlp.gate.weight", | |
| "model.llm.layers.7.mlp_norm.weight", | |
| "model.llm.layers.7.mlp_sconv.weight", | |
| "model.llm.layers.8.attn.attn_sink_proj.weight", | |
| "model.llm.layers.8.attn.gate_attn.weight", | |
| "model.llm.layers.8.attn.k_sconv.weight", | |
| "model.llm.layers.8.attn.q_l2_clip.scale", | |
| "model.llm.layers.8.attn.rel_l2_clip.scale", | |
| "model.llm.layers.8.attn.rel_logits_proj.proj", | |
| "model.llm.layers.8.attn.v_sconv.weight", | |
| "model.llm.layers.8.attn.wk_dv.weight", | |
| "model.llm.layers.8.attn.wo_ud.weight", | |
| "model.llm.layers.8.attn.wq_du.weight", | |
| "model.llm.layers.8.attn.wr_du.weight", | |
| "model.llm.layers.8.attn.wv_dv.weight", | |
| "model.llm.layers.8.attn_norm.weight", | |
| "model.llm.layers.8.attn_sconv.weight", | |
| "model.llm.layers.8.mlp.gate.weight", | |
| "model.llm.layers.8.mlp_norm.weight", | |
| "model.llm.layers.8.mlp_sconv.weight", | |
| "model.llm.layers.9.attn.attn_sink_proj.weight", | |
| "model.llm.layers.9.attn.gate_attn.weight", | |
| "model.llm.layers.9.attn.k_sconv.weight", | |
| "model.llm.layers.9.attn.q_l2_clip.scale", | |
| "model.llm.layers.9.attn.rel_l2_clip.scale", | |
| "model.llm.layers.9.attn.rel_logits_proj.proj", | |
| "model.llm.layers.9.attn.v_sconv.weight", | |
| "model.llm.layers.9.attn.wk_dv.weight", | |
| "model.llm.layers.9.attn.wo_ud.weight", | |
| "model.llm.layers.9.attn.wq_du.weight", | |
| "model.llm.layers.9.attn.wr_du.weight", | |
| "model.llm.layers.9.attn.wv_dv.weight", | |
| "model.llm.layers.9.attn_norm.weight", | |
| "model.llm.layers.9.attn_sconv.weight", | |
| "model.llm.layers.9.mlp.gate.weight", | |
| "model.llm.layers.9.mlp_norm.weight", | |
| "model.llm.layers.9.mlp_sconv.weight", | |
| "model.llm.norm.weight", | |
| "model.visual.backbone.embeddings.class_embedding", | |
| "model.visual.backbone.embeddings.cls_position_embedding", | |
| "model.visual.backbone.embeddings.patch_embedding.weight", | |
| "model.visual.backbone.embeddings.position_embedding", | |
| "model.visual.backbone.layers.0.ln_1.bias", | |
| "model.visual.backbone.layers.0.ln_1.weight", | |
| "model.visual.backbone.layers.0.ln_2.bias", | |
| "model.visual.backbone.layers.0.ln_2.weight", | |
| "model.visual.backbone.layers.0.ls_1.gamma", | |
| "model.visual.backbone.layers.0.ls_2.gamma", | |
| "model.visual.backbone.layers.0.mlp.c_fc.bias", | |
| "model.visual.backbone.layers.0.mlp.c_fc.weight", | |
| "model.visual.backbone.layers.0.mlp.c_proj.bias", | |
| "model.visual.backbone.layers.0.mlp.c_proj.weight", | |
| "model.visual.backbone.layers.0.self_attn.k_proj.bias", | |
| "model.visual.backbone.layers.0.self_attn.k_proj.weight", | |
| "model.visual.backbone.layers.0.self_attn.out_proj.bias", | |
| "model.visual.backbone.layers.0.self_attn.out_proj.weight", | |
| "model.visual.backbone.layers.0.self_attn.q_proj.bias", | |
| "model.visual.backbone.layers.0.self_attn.q_proj.weight", | |
| "model.visual.backbone.layers.0.self_attn.v_proj.bias", | |
| "model.visual.backbone.layers.0.self_attn.v_proj.weight", | |
| "model.visual.backbone.layers.1.ln_1.bias", | |
| "model.visual.backbone.layers.1.ln_1.weight", | |
| "model.visual.backbone.layers.1.ln_2.bias", | |
| "model.visual.backbone.layers.1.ln_2.weight", | |
| "model.visual.backbone.layers.1.ls_1.gamma", | |
| "model.visual.backbone.layers.1.ls_2.gamma", | |
| "model.visual.backbone.layers.1.mlp.c_fc.bias", | |
| "model.visual.backbone.layers.1.mlp.c_fc.weight", | |
| "model.visual.backbone.layers.1.mlp.c_proj.bias", | |
| "model.visual.backbone.layers.1.mlp.c_proj.weight", | |
| "model.visual.backbone.layers.1.self_attn.k_proj.bias", | |
| "model.visual.backbone.layers.1.self_attn.k_proj.weight", | |
| "model.visual.backbone.layers.1.self_attn.out_proj.bias", | |
| "model.visual.backbone.layers.1.self_attn.out_proj.weight", | |
| "model.visual.backbone.layers.1.self_attn.q_proj.bias", | |
| "model.visual.backbone.layers.1.self_attn.q_proj.weight", | |
| "model.visual.backbone.layers.1.self_attn.v_proj.bias", | |
| "model.visual.backbone.layers.1.self_attn.v_proj.weight", | |
| "model.visual.backbone.layers.2.ln_1.bias", | |
| "model.visual.backbone.layers.2.ln_1.weight", | |
| "model.visual.backbone.layers.2.ln_2.bias", | |
| "model.visual.backbone.layers.2.ln_2.weight", | |
| "model.visual.backbone.layers.2.ls_1.gamma", | |
| "model.visual.backbone.layers.2.ls_2.gamma", | |
| "model.visual.backbone.layers.2.mlp.c_fc.bias", | |
| "model.visual.backbone.layers.2.mlp.c_fc.weight", | |
| "model.visual.backbone.layers.2.mlp.c_proj.bias", | |
| "model.visual.backbone.layers.2.mlp.c_proj.weight", | |
| "model.visual.backbone.layers.2.self_attn.k_proj.bias", | |
| "model.visual.backbone.layers.2.self_attn.k_proj.weight", | |
| "model.visual.backbone.layers.2.self_attn.out_proj.bias", | |
| "model.visual.backbone.layers.2.self_attn.out_proj.weight", | |
| "model.visual.backbone.layers.2.self_attn.q_proj.bias", | |
| "model.visual.backbone.layers.2.self_attn.q_proj.weight", | |
| "model.visual.backbone.layers.2.self_attn.v_proj.bias", | |
| "model.visual.backbone.layers.2.self_attn.v_proj.weight", | |
| "model.visual.backbone.layers.3.ln_1.bias", | |
| "model.visual.backbone.layers.3.ln_1.weight", | |
| "model.visual.backbone.layers.3.ln_2.bias", | |
| "model.visual.backbone.layers.3.ln_2.weight", | |
| "model.visual.backbone.layers.3.ls_1.gamma", | |
| "model.visual.backbone.layers.3.ls_2.gamma", | |
| "model.visual.backbone.layers.3.mlp.c_fc.bias", | |
| "model.visual.backbone.layers.3.mlp.c_fc.weight", | |
| "model.visual.backbone.layers.3.mlp.c_proj.bias", | |
| "model.visual.backbone.layers.3.mlp.c_proj.weight", | |
| "model.visual.backbone.layers.3.self_attn.k_proj.bias", | |
| "model.visual.backbone.layers.3.self_attn.k_proj.weight", | |
| "model.visual.backbone.layers.3.self_attn.out_proj.bias", | |
| "model.visual.backbone.layers.3.self_attn.out_proj.weight", | |
| "model.visual.backbone.layers.3.self_attn.q_proj.bias", | |
| "model.visual.backbone.layers.3.self_attn.q_proj.weight", | |
| "model.visual.backbone.layers.3.self_attn.v_proj.bias", | |
| "model.visual.backbone.layers.3.self_attn.v_proj.weight", | |
| "model.visual.backbone.ln_pre.bias", | |
| "model.visual.backbone.ln_pre.weight", | |
| "model.visual.proj_in.weight", | |
| "model.visual.proj_out.weight" | |
| ], | |
| "weight_block_shape": [], | |
| "weight_transform": "cast" | |
| }, | |
| { | |
| "compute_dtype": "nvfp4", | |
| "output_dtype": "uint8", | |
| "required_activation_quantization": { | |
| "amax_type": "per_token", | |
| "block_size": 16, | |
| "format": "nvfp4_e2m1", | |
| "rounding": "nearest", | |
| "scale_dtype": "float8_e4m3fn", | |
| "scale_selection_rule": "fouroversix_mae" | |
| }, | |
| "tensors": [ | |
| "model.llm.layers.0.mlp.experts.w13_weight", | |
| "model.llm.layers.0.mlp.experts.w2_weight", | |
| "model.llm.layers.1.mlp.experts.w13_weight", | |
| "model.llm.layers.1.mlp.experts.w2_weight", | |
| "model.llm.layers.2.mlp.experts.w13_weight", | |
| "model.llm.layers.2.mlp.experts.w2_weight", | |
| "model.llm.layers.3.mlp.experts.w13_weight", | |
| "model.llm.layers.3.mlp.experts.w2_weight", | |
| "model.llm.layers.4.mlp.experts.w13_weight", | |
| "model.llm.layers.4.mlp.experts.w2_weight", | |
| "model.llm.layers.5.mlp.experts.w13_weight", | |
| "model.llm.layers.5.mlp.experts.w2_weight" | |
| ], | |
| "weight_block_shape": [ | |
| 1, | |
| 16 | |
| ], | |
| "weight_scale_format": "amax_denominator", | |
| "weight_transform": "nvfp4_nearest" | |
| }, | |
| { | |
| "compute_dtype": "mxfp8", | |
| "output_dtype": "float8_e4m3fn", | |
| "required_activation_quantization": { | |
| "block_size": 32, | |
| "format": "mxfp8_e4m3", | |
| "rounding": "nearest", | |
| "scale_dtype": "ue8m0" | |
| }, | |
| "tensors": [ | |
| "model.llm.layers.10.mlp.experts.w13_weight", | |
| "model.llm.layers.10.mlp.experts.w2_weight", | |
| "model.llm.layers.11.mlp.experts.w13_weight", | |
| "model.llm.layers.11.mlp.experts.w2_weight", | |
| "model.llm.layers.6.mlp.experts.w13_weight", | |
| "model.llm.layers.6.mlp.experts.w2_weight", | |
| "model.llm.layers.7.mlp.experts.w13_weight", | |
| "model.llm.layers.7.mlp.experts.w2_weight", | |
| "model.llm.layers.8.mlp.experts.w13_weight", | |
| "model.llm.layers.8.mlp.experts.w2_weight", | |
| "model.llm.layers.9.mlp.experts.w13_weight", | |
| "model.llm.layers.9.mlp.experts.w2_weight" | |
| ], | |
| "weight_block_shape": [ | |
| 32, | |
| 32 | |
| ], | |
| "weight_transform": "mxfp8_nearest" | |
| }, | |
| { | |
| "compute_dtype": "bfloat16", | |
| "fwd_output_dtype": "float32", | |
| "output_dtype": "bfloat16", | |
| "required_activation_quantization": null, | |
| "tensors": [ | |
| "model.llm.unembed.weight" | |
| ], | |
| "weight_block_shape": [], | |
| "weight_transform": "cast" | |
| } | |
| ] | |
| }, | |
| "mtp": { | |
| "parameter_groups": [ | |
| { | |
| "compute_dtype": "bfloat16", | |
| "output_dtype": "bfloat16", | |
| "required_activation_quantization": null, | |
| "tensors": [ | |
| "fuse_weight", | |
| "layers.0.dflare_norm.weight", | |
| "layers.0.input_layernorm.weight", | |
| "layers.0.mlp.down_proj.weight", | |
| "layers.0.mlp.gate_proj.weight", | |
| "layers.0.mlp.up_proj.weight", | |
| "layers.0.post_attention_layernorm.weight", | |
| "layers.0.self_attn.k_norm.weight", | |
| "layers.0.self_attn.k_norm_ctx.weight", | |
| "layers.0.self_attn.k_proj.weight", | |
| "layers.0.self_attn.k_proj_ctx.weight", | |
| "layers.0.self_attn.o_proj.weight", | |
| "layers.0.self_attn.q_norm.weight", | |
| "layers.0.self_attn.q_proj.weight", | |
| "layers.0.self_attn.v_proj.weight", | |
| "layers.0.self_attn.v_proj_ctx.weight", | |
| "layers.1.dflare_norm.weight", | |
| "layers.1.input_layernorm.weight", | |
| "layers.1.mlp.down_proj.weight", | |
| "layers.1.mlp.gate_proj.weight", | |
| "layers.1.mlp.up_proj.weight", | |
| "layers.1.post_attention_layernorm.weight", | |
| "layers.1.self_attn.k_norm.weight", | |
| "layers.1.self_attn.k_norm_ctx.weight", | |
| "layers.1.self_attn.k_proj.weight", | |
| "layers.1.self_attn.k_proj_ctx.weight", | |
| "layers.1.self_attn.o_proj.weight", | |
| "layers.1.self_attn.q_norm.weight", | |
| "layers.1.self_attn.q_proj.weight", | |
| "layers.1.self_attn.v_proj.weight", | |
| "layers.1.self_attn.v_proj_ctx.weight", | |
| "layers.2.dflare_norm.weight", | |
| "layers.2.input_layernorm.weight", | |
| "layers.2.mlp.down_proj.weight", | |
| "layers.2.mlp.gate_proj.weight", | |
| "layers.2.mlp.up_proj.weight", | |
| "layers.2.post_attention_layernorm.weight", | |
| "layers.2.self_attn.k_norm.weight", | |
| "layers.2.self_attn.k_norm_ctx.weight", | |
| "layers.2.self_attn.k_proj.weight", | |
| "layers.2.self_attn.k_proj_ctx.weight", | |
| "layers.2.self_attn.o_proj.weight", | |
| "layers.2.self_attn.q_norm.weight", | |
| "layers.2.self_attn.q_proj.weight", | |
| "layers.2.self_attn.v_proj.weight", | |
| "layers.2.self_attn.v_proj_ctx.weight", | |
| "layers.3.dflare_norm.weight", | |
| "layers.3.input_layernorm.weight", | |
| "layers.3.mlp.down_proj.weight", | |
| "layers.3.mlp.gate_proj.weight", | |
| "layers.3.mlp.up_proj.weight", | |
| "layers.3.post_attention_layernorm.weight", | |
| "layers.3.self_attn.k_norm.weight", | |
| "layers.3.self_attn.k_norm_ctx.weight", | |
| "layers.3.self_attn.k_proj.weight", | |
| "layers.3.self_attn.k_proj_ctx.weight", | |
| "layers.3.self_attn.o_proj.weight", | |
| "layers.3.self_attn.q_norm.weight", | |
| "layers.3.self_attn.q_proj.weight", | |
| "layers.3.self_attn.v_proj.weight", | |
| "layers.3.self_attn.v_proj_ctx.weight", | |
| "layers.4.dflare_norm.weight", | |
| "layers.4.input_layernorm.weight", | |
| "layers.4.mlp.down_proj.weight", | |
| "layers.4.mlp.gate_proj.weight", | |
| "layers.4.mlp.up_proj.weight", | |
| "layers.4.post_attention_layernorm.weight", | |
| "layers.4.self_attn.k_norm.weight", | |
| "layers.4.self_attn.k_norm_ctx.weight", | |
| "layers.4.self_attn.k_proj.weight", | |
| "layers.4.self_attn.k_proj_ctx.weight", | |
| "layers.4.self_attn.o_proj.weight", | |
| "layers.4.self_attn.q_norm.weight", | |
| "layers.4.self_attn.q_proj.weight", | |
| "layers.4.self_attn.v_proj.weight", | |
| "layers.4.self_attn.v_proj_ctx.weight", | |
| "markov_head.dflare_norm.weight", | |
| "markov_head.hidden_out.bias", | |
| "markov_head.hidden_out.weight", | |
| "markov_head.markov_w1.weight", | |
| "markov_head.rnn_proj.bias", | |
| "markov_head.rnn_proj.weight", | |
| "mask_embed", | |
| "norm.weight" | |
| ], | |
| "weight_block_shape": [], | |
| "weight_transform": "cast" | |
| } | |
| ] | |
| }, | |
| "quant_method": "inkling2" | |
| } | |