improve hf_to_nvfp4_safetensors

This commit is contained in:
zefie committed 2026-11-04 14:30:11 -05:00
1 parent b7ac8ed00c
commit baa9c09e0e
4 files changed
+172 -31

No files matched your search

+6
View File
@@ -7,6 +7,9 @@ from .quant import (
DEFAULT_EARLY_STOP_FLOOR_STALL,
DEFAULT_TARGET_LOSS,
DEFAULT_EARLY_STOP_STALL,
DEFAULT_LR_FLOOR,
DEFAULT_LR_PATIENCE,
DEFAULT_LR_THRESHOLD,
DEFAULT_SCALE_OPTIMIZATION,
DEFAULT_SCALE_REFINEMENT_ROUNDS,
FP4_LEVELS,
@@ -59,6 +62,9 @@ __all__ = [
"DEFAULT_EARLY_STOP_FLOOR_STALL",
"DEFAULT_TARGET_LOSS",
"DEFAULT_EARLY_STOP_STALL",
"DEFAULT_LR_FLOOR",
"DEFAULT_LR_PATIENCE",
"DEFAULT_LR_THRESHOLD",
"DEFAULT_SCALE_OPTIMIZATION",
"DEFAULT_SCALE_REFINEMENT_ROUNDS",
"FP4_LEVELS",