Solar-Open-100B-NotaMoeQuant-Int4
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.
Overview
- Base model: Solar-Open-100B
- Quantization: Int4 weight-only
- Packing format:
auto_round:auto_gptq(ensuring backend compatibility with PyTorch and vLLM) - Quantization group size: 128
- Supported tensor parallel sizes: {1,2}
- Hardware Requirements:
- Minimum: 2 x NVIDIA A100 (80GB)
License
This repository contains both model weights and code, which are licensed under different terms:
MODEL WEIGHTS (*.safetensors) Licensed under Upstage Solar License See: https://huggingface.co/upstage/Solar-Open-100B/blob/main/LICENSE
CODE (*.py, *.json, *.jinja files) Licensed under Apache License 2.0 See: https://www.apache.org/licenses/LICENSE-2.0
Performance
- English
| Solar-Open-100B | Nota MoE Quantization (Ours) | AutoRound | cyankiwi AWQ | |
|---|---|---|---|---|
| PPL (WikiText-2)↓ | 6.06 | 6.81 | 7.12 | 30.52 |
| PPL (C4)↓ | 20.37 | 20.84 | 20.94 | 50.16 |
| PIQA↑ | 82.37 | 82.75 | 82.05 | 78.94 |
| BoolQ↑ | 84.89 | 84.86 | 85.29 | 68.87 |
| ARC-E↑ | 87.25 | 86.48 | 85.77 | 83.12 |
| ARC-C↑ | 61.43 | 61.69 | 60.84 | 56.40 |
| TruthfulQA↑ | 59.25 | 60.14 | 59.18 | 52.38 |
| WinoGrande↑ | 76.09 | 75.77 | 75.77 | 68.59 |
- Korean
| Solar-Open-100B | Nota MoE Quantization (Ours) | AutoRound | cyankiwi AWQ | |
|---|---|---|---|---|
| HRM8K↑ | 81.52 | 80.68 | 81.56 | 32.67 |
| MMLU-ProX-Lite↑ | 55.44 | 51.84 | 51.26 | 6.19 |
| KoBEST↑ | 62.00 | 62.80 | 61.80 | 61.80 |
| CLiCK↑ | 71.33 | 70.03 | 69.77 | 51.18 |
- Model weigth memory footprint
| Solar-Open-100B | Nota MoE Quantization (Ours) | cyankiwi AWQ |
|---|---|---|
| 191.2 GB | 51.9 GB | 57.0 GB |
- Note
- ↑ / ↓ denote the direction of improvement: higher is better (↑), lower is better (↓).
- Cyankiwi AWQ is a publicly available INT4 (4-bit AWQ) quantized version of Solar-Open-100B
- Because we used a smaller thinking budget, the results for HRM8K and CLiCK are slightly lower than the numbers reported in the original Solar-Open-100B repository.
- Memory refers to the pure VRAM footprint occupied only by the model weights.
Inference
Transformers
Install the required dependencies:
pip install -U transformers kernels torch accelerate auto-round==0.8.0
Run inference with the following code:
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
MODEL_ID = "nota-ai/Solar-Open-100B-NotaMoEQuant-Int4"
# Load model and tokenizer
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
model = AutoModelForCausalLM.from_pretrained(
pretrained_model_name_or_path=MODEL_ID,
torch_dtype=torch.bfloat16,
device_map="auto",
trust_remote_code=True,
)
# Prepare input
messages = [{"role": "user", "content": "who are you?"}]
inputs = tokenizer.apply_chat_template(
messages,
tokenize=True,
add_generation_prompt=True,
return_dict=True,
return_tensors="pt",
)
inputs = inputs.to(model.device)
# Generate response
generated_ids = model.generate(
**inputs,
max_new_tokens=4096,
temperature=0.8,
top_p=0.95,
top_k=50,
do_sample=True,
)
generated_text = tokenizer.decode(generated_ids[0][inputs.input_ids.shape[1] :])
print(generated_text)
vLLM
Create and activate a Python virtual environment
uv venv --python 3.12 --seed
source .venv/bin/activate
Install Solar Open's optimized vLLM
VLLM_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" \
VLLM_USE_PRECOMPILED=1 \
uv pip install git+https://github.com/UpstageAI/vllm.git@v0.12.0-solar-open
Start the vLLM server (For 2 GPUs)
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
vllm serve nota-ai/Solar-Open-100B-NotaMoEQuant-Int4 \
--trust-remote-code \
--enable-auto-tool-choice \
--tool-call-parser solar_open \
--reasoning-parser solar_open \
--logits-processors vllm.model_executor.models.parallel_tool_call_logits_processor:ParallelToolCallLogitsProcessor \
--logits-processors vllm.model_executor.models.solar_open_logits_processor:SolarOpenTemplateLogitsProcessor \
--tensor-parallel-size 2 \
--max-num-seqs 64 \
--gpu-memory-utilization 0.8
- Downloads last month
- -