Fix:

- Remove LSA parameters from parameters dict when the sanity check for block_id_and_param_type fails and lsa is disabled
fraunhoferhhi · Dec 16, 2022 · cf7ce60 · cf7ce60
1 parent da45cfc
commit cf7ce60
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -57,7 +57,7 @@ source env/bin/activate
 
 **Note**: For further information on how to set up a virtual python environment (also on **Windows**) refer to https://docs.python.org/3/library/venv.html .
 
-When successfully installed, the software outputs the line : "Successfully installed NNC-0.1.1"
+When successfully installed, the software outputs the line : "Successfully installed NNC-0.1.2"
 
 ### Importing the main module
 

diff --git a/nnc/compression.py b/nnc/compression.py
@@ -232,6 +232,9 @@ def compress_model( model_path_or_object,
             if model_executer:
                 model_executer.model = model_executer.original_model
                 del model_executer.original_model
+            for key in model_parameters.keys():
+                if "weight_scaling" in key:
+                    del model_parameters[key]
 
 
     bitstream = compress(   model_parameters,

diff --git a/setup.py b/setup.py
@@ -49,7 +49,7 @@
 from setuptools.command.build_ext import build_ext
 import setuptools
 
-__version__ = '0.1.1'
+__version__ = '0.1.2'
 
 
 class get_pybind_include(object):