API Reference¶
This section provides detailed API documentation for all public classes and functions in Fujitsu One Compression (OneComp).
Top-level Exports¶
The following are available directly from import onecomp:
| Name | Type | Description |
|---|---|---|
Runner |
Class | Main entry point for quantization |
ModelConfig |
Class | Model and tokenizer configuration |
QEPConfig |
Class | QEP configuration |
LPCDConfig |
Class | LPCD configuration |
AutoBitQuantizer |
Class | AutoBit mixed-precision quantizer (ILP) |
GPTQ |
Class | GPTQ quantizer |
RTN |
Class | RTN quantizer |
DBF |
Class | DBF quantizer |
JointQ |
Class | JointQ quantizer |
QUIP |
Class | QuIP quantizer |
ARB |
Class | ARB quantizer |
CQ |
Class | CQ quantizer |
QBB |
Class | QBB quantizer |
Onebit |
Class | 1-bit quantizer |
RotatedModelConfig |
Class | ModelConfig for rotation-preprocessed models |
QuantizedModelLoader |
Class | Loader for saved quantized models |
load_quantized_model |
Function | Shortcut for QuantizedModelLoader.load_quantized_model |
load_quantized_model_pt |
Function | Shortcut for QuantizedModelLoader.load_quantized_model_pt |
prepare_rotated_model |
Function | Rotation preprocessing pipeline |
setup_logger |
Function | Configure logging output |
Evaluation (onecomp-eval, vLLM + MT-Bench / throughput) is documented separately:
Evaluation user guide and Evaluation API.
Module Structure¶
onecomp/
runner.py # Runner class (includes auto_run)
cli.py # CLI entry point (onecomp command)
__main__.py # python -m onecomp support
model_config.py # ModelConfig class
rotated_model_config.py # RotatedModelConfig class
pre_process/ # Rotation preprocessing pipeline
prepare_rotated_model.py # prepare_rotated_model()
qep/ # QEP module
_qep_config.py # QEPConfig dataclass
lpcd/ # LPCD module
_lpcd_config.py # LPCDConfig dataclass
quantizer/ # Quantizer implementations
_quantizer.py # Quantizer base class, QuantizationResult
autobit/ # AutoBit (ILP mixed-precision)
gptq/ # GPTQ
dbf/ # DBF
rtn/ # RTN
jointq/ # JointQ
...
quantized_model_loader.py # QuantizedModelLoader
log.py # setup_logger
utils/ # Calibration, perplexity, accuracy utilities
eval/ # onecomp-eval harness (see api/eval.md)
schema.py # EvalConfig, TaskResult, per-eval configs
orchestrator/ # vLLM server, subprocess dispatch, aggregation
evals/ # mt_bench, throughput, ...