Inkling-0.92B-MTP / README.md
soyrsoyr's picture
simplify model card to recipe and usage
d3c7659 verified
|
Raw History Blame Contribute Delete
637 Bytes
metadata
library_name: transformers
license: apache-2.0
tags:
  - llm-compressor
  - tiny-model
  - mtp
  - pr-3225
  - test-fixture

Inkling-0.92B-MTP

Small BF16 test model using thinkingmachines/Inkling, with a toy-trained backbone and synthetic MTP weights.

Load the model

This fixture has no published quantized variant.

import torch
from transformers import InklingForConditionalGeneration

MODEL_ID = "inference-optimization/Inkling-0.92B-MTP"
model = InklingForConditionalGeneration.from_pretrained(
    MODEL_ID, dtype=torch.bfloat16, attn_implementation="eager",
)