@@ -323,7 +323,7 @@ def __init__(
323
323
if isinstance (ignored_scope , nncf .IgnoredScope ):
324
324
ignored_scope = ignored_scope .__dict__
325
325
self .ignored_scope = ignored_scope
326
- self .init_kwargs = kwargs
326
+ self .kwargs = kwargs
327
327
328
328
def post_init (self ):
329
329
try :
@@ -344,9 +344,9 @@ def clone(self):
344
344
return copy .deepcopy (self )
345
345
346
346
def to_dict (self ) -> Dict [str , Any ]:
347
- # Unpack init kwargs back into kwargs
347
+ # Unpack kwargs dict
348
348
result = super ().to_dict ()
349
- result = result | result .pop ("init_kwargs " , {})
349
+ result = result | result .pop ("kwargs " , {})
350
350
return result
351
351
352
352
@@ -634,7 +634,7 @@ def to_nncf_dict(self) -> Dict[str, Any]:
634
634
"gptq" : self .gptq ,
635
635
"lora_correction" : self .lora_correction ,
636
636
"backup_mode" : backup_mode ,
637
- ** self .init_kwargs ,
637
+ ** self .kwargs ,
638
638
}
639
639
return result
640
640
@@ -783,9 +783,9 @@ def to_nncf_dict(self) -> Dict[str, Any]:
783
783
Returns a dictionary with the variables that are ready to use for nncf.compress_weights() call.
784
784
"""
785
785
786
- # Merge advanced parameters from init_kwargs if they were provided
787
- init_kwargs_copy = copy .deepcopy (self .init_kwargs )
788
- advanced_parameters = init_kwargs_copy .pop ("advanced_parameters" , nncf .AdvancedQuantizationParameters ())
786
+ # Merge advanced parameters from kwargs if they were provided
787
+ kwargs_copy = copy .deepcopy (self .kwargs )
788
+ advanced_parameters = kwargs_copy .pop ("advanced_parameters" , nncf .AdvancedQuantizationParameters ())
789
789
advanced_parameters .overflow_fix = nncf .OverflowFix (self .overflow_fix )
790
790
if self .smooth_quant_alpha :
791
791
advanced_parameters .smooth_quant_alphas .matmul = self .smooth_quant_alpha
@@ -805,7 +805,7 @@ def to_nncf_dict(self) -> Dict[str, Any]:
805
805
"model_type" : model_type ,
806
806
"ignored_scope" : self .get_ignored_scope_instance (),
807
807
"advanced_parameters" : advanced_parameters ,
808
- ** init_kwargs_copy ,
808
+ ** kwargs_copy ,
809
809
}
810
810
811
811
0 commit comments