import transformers
import torch

model_id = "miike-ai/r1-12b"

pipeline = transformers.pipeline(
    "text-generation",
    model=model_id,
    model_kwargs={"torch_dtype": torch.bfloat16},
    device_map="auto",
)

messages = [
    {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
    {"role": "user", "content": "Who are you?"},
]

outputs = pipeline(
    messages,
    max_new_tokens=8192,
)
print(outputs[0]["generated_text"][-1])
Downloads last month
472
Safetensors
Model size
11.5B params
Tensor type
BF16
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.

Model tree for miike-ai/r1-12b

Finetuned
(55)
this model
Adapters
1 model
Quantizations
1 model