This repository provides
Upstage’s flagship model, Solar-Open-100B, packaged with
Nota AI’s proprietary quantization technique specifically developed for Mixture-of-Experts (MoE)-based LLMs. Unlike conventional quantization methods, this approach incorporates a novel method designed to mitigate representation distortion that can occur when experts are mixed under quantization in MoE architectures.
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-NotaMoEQuant-Int4"
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-NotaMoEQuant-Int4 \
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