add reuse flag to weights quantizers, change workflow to mctq nightly

Author: eladco

.github/workflows/nightly.yml

@@ -31,8 +31,11 @@ jobs:
       - name: Publish nightly
         run: |
           twine upload --repository pypi dist/* --verbose -u __token__ -p ${{ secrets.PYPI_API_SSI_DEV_NIGHTLY_KEY }}
+      - name: Edit requirements mct-quantizers name # Remove the entire line in requirements.txt so no need special care for the specified version
+        run: |
+          sed -i "/mct-quantizers/c\mct-quantizers-nightly" requirements.txt
       - name: Post publish import test
         run: |
           pip install mct-nightly tensorflow torch
           version=$(python -c 'import model_compression_toolkit; print(model_compression_toolkit.__version__)')
-          echo $version
+          echo $version

model_compression_toolkit/exporter/model_wrapper/pytorch/builder/fully_quantized_model_builder.py

@@ -45,6 +45,10 @@ def fully_quantized_wrapper(node: common.BaseNode,
         """
         weight_quantizers, _ = fw_impl.get_inferable_quantizers(node)
         if len(weight_quantizers) > 0:
+            # Set reuse for weight quantizers if node is reused
+            for _, quantizer in weight_quantizers.items():
+                if node.reuse_group:
+                    quantizer.enable_reuse_quantizer()
             # for positional weights we need to extract the weight's value.
             weights_values = {attr: fw_impl.to_tensor(node.get_weights_by_keys(attr))
                               for attr in weight_quantizers if isinstance(attr, int)}