Buckets:
| language: | |
| - en | |
| - ko | |
| library_name: transformers | |
| license: other | |
| license_name: upstage-solar-license | |
| pipeline_tag: text-generation | |
| tags: | |
| - upstage | |
| - solar | |
| - moe | |
| - 100b | |
| - llm | |
| - nota | |
| - quantization | |
| # **Solar-Open-100B-NotaMoeQuant-Int4** | |
| This repository provides **Upstage’s flagship model, [Solar-Open-100B](https://huggingface.co/upstage/Solar-Open-100B)**, packaged with [**Nota AI**](https://www.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](https://huggingface.co/upstage/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: | |
| 1. MODEL WEIGHTS (*.safetensors) | |
| Licensed under **Upstage Solar License** | |
| See: https://huggingface.co/upstage/Solar-Open-100B/blob/main/LICENSE | |
| 2. 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](cyankiwi/Solar-Open-100B-AWQ-4bit) | |
| - 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: | |
| ```bash | |
| pip install -U transformers kernels torch accelerate auto-round==0.8.0 | |
| ``` | |
| Run inference with the following code: | |
| ```python | |
| 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 | |
| ```bash | |
| uv venv --python 3.12 --seed | |
| source .venv/bin/activate | |
| ``` | |
| Install Solar Open's optimized vLLM | |
| ```bash | |
| 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) | |
| ```bash | |
| 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 | |
| ``` | |
Xet Storage Details
- Size:
- 6.13 kB
- Xet hash:
- 09f82689bd205a1086462843453973043a99d01abb5c61ca741ac4ab186a8dd7
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.