minor

Signed-off-by: realAsma <[email protected]>

Author: realAsma

modelopt/torch/quantization/plugins/transformers_trainer.py

@@ -191,7 +191,8 @@ def _restore_modelopt_state_with_weights(self):
         modelopt_state = torch.load(self._modelopt_state_path, weights_only=False)
         modelopt_weights = modelopt_state.pop("modelopt_state_weights", None)
         restore_from_modelopt_state(self.model, modelopt_state)
-        set_quantizer_state_dict(self.model, modelopt_weights)
+        if modelopt_weights is not None:
+            set_quantizer_state_dict(self.model, modelopt_weights)
         print_rank_0("Restored modelopt state with weights.")
 
     def _quantize_model(self):