To check if the param_encodings field is present in the encoding file…

… passed to set_and_freeze_param_encoding api and use param_encodings dict from that.

Signed-off-by: Rishabh Thakur <[email protected]>

TrainingExtensions/torch/src/python/aimet_torch/quantsim.py

@@ -1630,6 +1630,10 @@ def set_and_freeze_param_encodings(self, encoding_path: str):
         with open(encoding_path) as json_file:
             param_encodings = json.load(json_file)
 
+            # In case of full encoding file is provided, param_encoding is nested inside the top level encoding structure
+            if 'param_encodings' in param_encodings:
+                param_encodings = param_encodings['param_encodings']
+
         for name, quant_module in self.model.named_modules():
             if isinstance(quant_module, QcQuantizeWrapper):
                 quant_module.set_param_encoding(name, param_encodings)