improve quantization
This commit is contained in:
1 parent
0c49394911
commit
f1e394f1c9
5 files changed
+335
-251
No files matched your search
@@ -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",
|
||||
|
||||
Reference in new issue
Block a user