Views
No views yet

FP8 Block (FP8 weights and activations, per-block scaling with E8M0 scale format)compressed-tensors

# pip
pip install "vllm>=0.18.0"VLLM_USE_DEEP_GEMM=1 vllm serve arcee-ai/Trinity-Mini-FP8-Block \
--trust-remote-code \
--max-model-len 4096 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_r1 \
--tool-call-parser hermesvllm serve arcee-ai/Trinity-Mini-FP8-Block \
--trust-remote-code \
--max-model-len 4096 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_r1 \
--tool-call-parser hermesmain transformers branchgit clone https://github.com/huggingface/transformers.git
cd transformers
# pip
pip install '.[torch]'
# uv
uv pip install '.[torch]'1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_id = "arcee-ai/Trinity-Mini-FP8-Block"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto",
10 trust_remote_code=True
11)
12
13messages = [
14 {"role": "user", "content": "Who are you?"},
15]
16
17input_ids = tokenizer.apply_chat_template(
18 messages,
19 add_generation_prompt=True,
20 return_tensors="pt"
21).to(model.device)
22
23outputs = model.generate(
24 input_ids,
25 max_new_tokens=256,
26 do_sample=True,
27 temperature=0.15,
28 top_k=50,
29 top_p=0.75,
30 min_p=0.06
31)
32
33response = tokenizer.decode(outputs[0], skip_special_tokens=True)
34print(response)curl -X POST "https://openrouter.ai/v1/chat/completions" \
-H "Authorization: Bearer $OPENROUTER_API_KEY" \
-H "Content-Type: application/json" \
-d '{
"model": "arcee-ai/trinity-mini",
"messages": [
{
"role": "user",
"content": "What are some fun things to do in New York?"
}
]
}'