Add default qskip_large_mag_layers to qcfg

Signed-off-by: Andrea Fasoli <[email protected]>

Author: andrea-fasoli

fms_mo/utils/qconfig_utils.py

@@ -297,6 +297,7 @@ def qconfig_init(recipe: str = None, args: Any = None) -> dict:
     # ways to control which layers to be quantized/skipped
     qcfg["qlayer_name_pattern"] = []
     qcfg["qskip_layer_name"] = []
+    qcfg["qskip_large_mag_layers"] = False
     qcfg["qspecial_layers"] = {}
 
     # settings about quantizing bmm/matmul
@@ -538,9 +539,9 @@ def add_wanted_defaults_to_config(config: dict, minimal: bool = True) -> None:
 
 def qconfig_save(
     qcfg: dict,
-    recipe: str | None = None,
+    recipe: str = None,
     minimal: bool = True,
-    fname: str | Path = "qcfg.json",
+    fname="qcfg.json",
 ) -> None:
     """
     Try to save qcfg into a JSON file (or use .pt format if something really can't be text-only).
@@ -550,8 +551,8 @@ def qconfig_save(
     Args:
         qcfg (dict): Quantized config.
         recipe (str, optional): String name for a save recipe. Defaults to None.
-        minimal (bool): Save a minimal quantized config. Defaults to True.
-        fname (str): File name to save quantized config. Defaults to "qcfg.json".
+        minimal (bool, optional): Save a minimal quantized config. Defaults to True.
+        fname (str, optional): File name to save quantized config. Defaults to "qcfg.json".
     """
 
     # First check in qcfg for added save list
@@ -598,7 +599,6 @@ def qconfig_save(
         warnings.warn(message, UserWarning)
     with open(fname, "w", encoding="utf-8") as outfile:
         json.dump(temp_qcfg, outfile, indent=4)
-    logger.info(f"Quantization configuration saved to {fname}")
 
 
 def qconfig_load(fname: str = "qcfg.json") -> dict: