NexusCoder / nexus /optim /__init__.py
AdminReal's picture
Import NexusCoder from github.com/mhieuhonda/NexusCoder
eca5751 verified
Raw
History Blame Contribute Delete
563 Bytes
"""
Nexus Optim Module - v0.2 NEW
=============================
Tối ưu model cho inference và training.
Modules:
- quantization: INT8/INT4/FP8 quantization
- lora: LoRA / QLoRA efficient fine-tuning
- distillation: Knowledge distillation
- pruning: Structured / unstructured pruning
"""
from .quantization import Quantizer
from .lora import LoRAConfig, LoRALinear, apply_lora
from .distillation import Distiller
from .pruning import Pruner
__all__ = [
"Quantizer",
"LoRAConfig",
"LoRALinear",
"apply_lora",
"Distiller",
"Pruner",
]