This repository provides an FP8-quantized version of
Upstage’s flagship model, Solar-Open-100B.
This repository contains both model weights and code,
which are licensed under different terms:
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4MODEL_ID = "nota-ai/Solar-Open-100B-Nota-FP8"
5
6# Load model and tokenizer
7tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
8
9model = AutoModelForCausalLM.from_pretrained(
10 pretrained_model_name_or_path=MODEL_ID,
11 torch_dtype=torch.bfloat16,
12 device_map="auto",
13 trust_remote_code=True,
14)
15
16# Prepare input
17messages = [{"role": "user", "content": "who are you?"}]
18inputs = tokenizer.apply_chat_template(
19 messages,
20 tokenize=True,
21 add_generation_prompt=True,
22 return_dict=True,
23 return_tensors="pt",
24)
25inputs = inputs.to(model.device)
26
27# Generate response
28generated_ids = model.generate(
29 **inputs,
30 max_new_tokens=4096,
31 temperature=0.8,
32 top_p=0.95,
33 top_k=50,
34 do_sample=True,
35)
36generated_text = tokenizer.decode(generated_ids[0][inputs.input_ids.shape[1] :])
37print(generated_text)
1uv venv --python 3.12 --seed
2source .venv/bin/activate
1VLLM_PRECOMPILED_WHEEL_LOCATION="https://github.com/vllm-project/vllm/releases/download/v0.12.0/vllm-0.12.0-cp38-abi3-manylinux_2_31_x86_64.whl" \
2VLLM_USE_PRECOMPILED=1 \
3uv pip install git+https://github.com/UpstageAI/vllm.git@v0.12.0-solar-open
1PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
2vllm serve nota-ai/Solar-Open-100B-Nota-FP8 \
3 --trust-remote-code \
4 --enable-auto-tool-choice \
5 --tool-call-parser solar_open \
6 --reasoning-parser solar_open \
7 --logits-processors vllm.model_executor.models.parallel_tool_call_logits_processor:ParallelToolCallLogitsProcessor \
8 --logits-processors vllm.model_executor.models.solar_open_logits_processor:SolarOpenTemplateLogitsProcessor \
9 --tensor-parallel-size 2 \
10 --max-num-seqs 64 \
11 --gpu-memory-utilization 0.8