Skip to main content

Production-grade model quantization SDK for enterprise custom models (AWQ, GGUF, and CoreML)

Project description

Qwodel - Production-Grade Model Quantization

Python 3.9+ License: MIT Code style: ruff

Qwodel is a production-ready Python package for model quantization across multiple backends (AWQ, GGUF, CoreML). It provides a unified, intuitive API for quantizing large language models with minimal code.

Features

  • Unified API - Simple interface across all quantization backends
  • Multiple Backends - AWQ (GPU), GGUF (CPU), CoreML (Apple devices)
  • Optional Dependencies - Install only what you need
  • CLI & Python API - Use via command line or programmatically
  • Type Safe - Full type hints and mypy validation
  • Well Documented - Comprehensive docs with examples

Quick Start

Installation

Quick Install (All Backends)

pip install qwodel[all]

This installs all backends (GGUF, AWQ, CoreML) with PyTorch 2.1.2 (CPU version).

GPU Support (for AWQ only)

If you need GPU quantization with AWQ, install PyTorch with CUDA first:

# 1. Install PyTorch with CUDA 12.1
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121

# 2. Install qwodel
pip install qwodel[all]

Note: GGUF and CoreML work perfectly fine with CPU-only PyTorch!

Individual Backends

# GGUF only (CPU quantization - most popular!)
pip install qwodel[gguf]

# AWQ only (GPU quantization)
pip install qwodel[awq]

# CoreML only (Apple devices)
pip install qwodel[coreml]

Local Development

# Clone and install locally
cd /path/to/qwodel
pip install -e .[all]

Python API

from qwodel import Quantizer

# Create quantizer
quantizer = Quantizer(
    backend="gguf",
    model_path="meta-llama/Llama-2-7b-hf",
    output_dir="./quantized"
)

# Quantize model
output_path = quantizer.quantize(format="Q4_K_M")
print(f"Quantized model saved to: {output_path}")

CLI

# Quantize a model
qwodel quantize \
    --backend gguf \
    --format Q4_K_M \
    --model meta-llama/Llama-2-7b-hf \
    --output ./quantized

# List available formats
qwodel list-formats --backend gguf

Supported Backends

GGUF (CPU Quantization)

  • Use Case: CPU inference, broad compatibility
  • Formats: Q4_K_M, Q8_0, Q2_K, Q5_K_M, and more
  • Best For: Most users, CPU-based deployment

AWQ (GPU Quantization)

  • Use Case: NVIDIA GPU inference
  • Formats: INT4
  • Best For: GPU deployments, maximum speed
  • Requires: CUDA 12.1+

CoreML (Apple Devices)

  • Use Case: iOS, macOS, iPadOS deployment
  • Formats: FLOAT16, INT8, INT4
  • Best For: Apple device deployment

Examples

Batch Processing

from qwodel import quantize

models = ["meta-llama/Llama-2-7b-hf", "meta-llama/Llama-2-13b-hf"]

for model in models:
    quantize(
        model_path=model,
        backend="gguf",
        format="Q4_K_M",
        output_dir="./quantized"
    )

Custom Progress Callback

from qwodel import Quantizer

def progress_handler(progress: int, stage: str, message: str):
    print(f"[{progress}%] {stage}: {message}")

quantizer = Quantizer(
    backend="gguf",
    model_path="./my-model",
    output_dir="./output",
    progress_callback=progress_handler
)

quantizer.quantize(format="Q4_K_M")

Documentation

Contributing

We welcome contributions! Please see CONTRIBUTING.md for guidelines.

License

MIT License - see LICENSE for details.

Acknowledgments

Qwodel builds upon the excellent work of:

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

qwodel-0.0.10.tar.gz (208.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

qwodel-0.0.10-py3-none-any.whl (208.7 kB view details)

Uploaded Python 3

File details

Details for the file qwodel-0.0.10.tar.gz.

File metadata

  • Download URL: qwodel-0.0.10.tar.gz
  • Upload date:
  • Size: 208.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.12

File hashes

Hashes for qwodel-0.0.10.tar.gz
Algorithm Hash digest
SHA256 819fd1184fe9e3ace0a745be74221c8a5607b81d8570154764daa64c8094dea4
MD5 93d428c061e452639a2b01b14c137208
BLAKE2b-256 b723a1fd4a0544c96f729295f9a22cc459c7b6a886eb39f50181f2da758e6a6a

See more details on using hashes here.

File details

Details for the file qwodel-0.0.10-py3-none-any.whl.

File metadata

  • Download URL: qwodel-0.0.10-py3-none-any.whl
  • Upload date:
  • Size: 208.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.12

File hashes

Hashes for qwodel-0.0.10-py3-none-any.whl
Algorithm Hash digest
SHA256 d87d56e0e09d0ac2933574eea58c8bcf27990e07fe845c05e424a6ff880989b7
MD5 d04d4a60fb33a00bb49311fe86b808e6
BLAKE2b-256 3a4767b86629dd6a232d9fd0a594b2395e2734897f0cd1dfcd5b0921402364b5

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page