improve quantization

This commit is contained in:
zefie committed 2026-11-01 18:30:02 -05:00
1 parent 0c49394911
commit f1e394f1c9
5 files changed
+335 -251

No files matched your search

+12
View File
@@ -44,6 +44,12 @@ from .quant import (
scale_bake_notice,
ue4m3_bits,
ue4m3_bits_torch,
coordinate_descent_refine,
coordinate_descent_refine_torch,
sweep_per_tensor_scale,
sweep_per_tensor_scale_torch,
get_ue4m3_lut,
get_fp4_torch_constants,
)
__all__ = [
@@ -84,6 +90,12 @@ __all__ = [
"quantize_and_pack",
"refine_codes_for_scale",
"refine_codes_for_scale_torch",
"coordinate_descent_refine",
"coordinate_descent_refine_torch",
"sweep_per_tensor_scale",
"sweep_per_tensor_scale_torch",
"get_ue4m3_lut",
"get_fp4_torch_constants",
"report_reconstruction",
"require_quant_settings",
"resolve_optimizer",