API change

vshampor · Apr 14, 2023 · a183142 · a183142
1 parent ef2d46c
commit a183142
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nncf/torch/model_creation.py b/nncf/torch/model_creation.py
@@ -53,7 +53,8 @@ def create_compressed_model(model: Module,
                             dummy_forward_fn: Callable[[Module], Any] = None,
                             wrap_inputs_fn: Callable[[Tuple, Dict], Tuple[Tuple, Dict]] = None,
                             wrap_outputs_fn: Callable[[Tuple, Dict], Tuple[Tuple, Dict]] = None,
-                            dump_graphs=True) \
+                            dump_graphs=True,
+                            new_arg = False) \
     -> Tuple[CompressionAlgorithmController, NNCFNetwork]:
     """
     The main function used to produce a model ready for compression fine-tuning from an original PyTorch