Text Generation
Transformers
Safetensors
English
Chinese
edgerock
rock-matrix
awq
int4
4-bit precision
quantized
reasoning
constant-memory
memory-efficient
edge-ai
custom_code
conversational
Instructions to use hkust-aaron/EdgeRock-4B-AWQ with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use hkust-aaron/EdgeRock-4B-AWQ with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="hkust-aaron/EdgeRock-4B-AWQ", trust_remote_code=True) messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("hkust-aaron/EdgeRock-4B-AWQ", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use hkust-aaron/EdgeRock-4B-AWQ with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "hkust-aaron/EdgeRock-4B-AWQ" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "hkust-aaron/EdgeRock-4B-AWQ", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/hkust-aaron/EdgeRock-4B-AWQ
- SGLang
How to use hkust-aaron/EdgeRock-4B-AWQ with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "hkust-aaron/EdgeRock-4B-AWQ" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "hkust-aaron/EdgeRock-4B-AWQ", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "hkust-aaron/EdgeRock-4B-AWQ" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "hkust-aaron/EdgeRock-4B-AWQ", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use hkust-aaron/EdgeRock-4B-AWQ with Docker Model Runner:
docker model run hf.co/hkust-aaron/EdgeRock-4B-AWQ
Download conversion.json from hkust-aaron/EdgeRock-4B-AWQ: direct link, hf CLI and curl.
- Browser
- Download file 10.2 kB
-
https://huggingface.co/hkust-aaron/EdgeRock-4B-AWQ/resolve/main/conversion.json
- Command line
-
hf download hf://hkust-aaron/EdgeRock-4B-AWQ/conversion.json
-
curl -L -o conversion.json https://huggingface.co/hkust-aaron/EdgeRock-4B-AWQ/resolve/main/conversion.json
10.2 kB
| { | |
| "method": "awq", | |
| "recipe": { | |
| "quant_cfg": [ | |
| { | |
| "quantizer_name": "*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*weight_quantizer", | |
| "cfg": { | |
| "num_bits": 4, | |
| "block_sizes": { | |
| "-1": 128, | |
| "type": "static" | |
| } | |
| } | |
| }, | |
| { | |
| "quantizer_name": "*input_quantizer", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*block_sparse_moe.gate*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*linear_attn.conv1d*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*self_attention.conv1d*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*linear_attn.in_proj_a*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*linear_attn.in_proj_b*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*lm_head*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*mixer.conv1d*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*mlp.gate.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*mlp.shared_expert_gate.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*output_layer*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*proj_out.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*router*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "mtp.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "output.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*embed_vision*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*vision_tower*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*visual*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*vision_model*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*multi_modal_projector*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*", | |
| "parent_class": "nn.BatchNorm1d", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*", | |
| "parent_class": "nn.BatchNorm2d", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*", | |
| "parent_class": "nn.BatchNorm3d", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*", | |
| "parent_class": "nn.LeakyReLU", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "*", | |
| "parent_class": "nn.Embedding", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.0.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.0.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.1.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.1.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.2.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.2.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.3.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.3.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.4.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.4.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.5.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.5.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.6.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.6.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.7.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.7.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.8.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.8.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.9.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.9.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.10.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.10.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.11.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.11.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.12.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.12.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.13.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.13.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.14.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.14.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.15.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.15.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.16.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.16.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.17.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.17.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.18.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.18.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.19.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.19.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.20.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.20.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.21.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.21.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.22.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.22.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.23.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.23.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.24.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.24.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.25.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.25.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.26.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.26.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.27.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.27.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.28.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.28.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.29.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.29.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.30.linear_attn.in_proj_b.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.30.linear_attn.in_proj_a.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.31.memory.admission_down.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "model.layers.31.memory.admission_up.*", | |
| "enable": false | |
| }, | |
| { | |
| "quantizer_name": "lm_head.*", | |
| "enable": false | |
| } | |
| ], | |
| "algorithm": { | |
| "method": "awq_lite", | |
| "alpha_step": 0.1 | |
| } | |
| }, | |
| "quantized_modules": 256, | |
| "maximum_weight_repacking_error": 0.0, | |
| "maximum_kernel_relative_error": 0.0049504950495049506, | |
| "weight_bytes": 3398133304, | |
| "seconds": 353.6774011319503, | |
| "versions": { | |
| "torch": "2.11.0", | |
| "transformers": "5.7.0.dev0", | |
| "nvidia-modelopt": "0.47.0", | |
| "triton": "3.6.0", | |
| "accelerate": "1.13.0", | |
| "safetensors": "0.7.0", | |
| "huggingface-hub": "1.5.0" | |
| }, | |
| "hardware": "NVIDIA H800" | |
| } | |