remove model as arg for _replace_initializer() (#1184) (#1187)

KSGulin · web-flow · commit 219e6680cdda · 2022-11-29T15:59:12.000Z
diff --git a/src/sparseml/pytorch/sparsification/quantization/quantize_qat_export.py b/src/sparseml/pytorch/sparsification/quantization/quantize_qat_export.py
@@ -259,7 +259,7 @@ def _replace_initializer(init_old, init_new):
     for init in model.graph.initializer:
         if init.data_type == 3:  # int8 dtype
             init_uint8 = _cast_init_int8_to_uint8(init)
-            _replace_initializer(model, init, init_uint8)
+            _replace_initializer(init, init_uint8)
 
 
 def _delete_repeated_qat_blocks(model: ModelProto):
