Apply suggestions from code review

Signed-off-by: Hubert Zhang <[email protected]>

Author: HubertZhang

checkpoint_engine/worker.py

@@ -172,17 +172,17 @@ def _load_weights(weights: _WEIGHTS_TYPE):
             self.model_runner.model.load_weights(weights)
             # Load drafter model weights if MTP/speculative decoding is enabled
             if (
-                getattr(self.model_runner, "drafter", None) is not None
-                and getattr(self.model_runner.drafter, "model", None) is not None
+                hasattr(self.model_runner, "drafter")
+                and hasattr(self.model_runner.drafter, "model")
             ):
                 self.model_runner.drafter.model.load_weights(weights=weights)
 
         def _post_hook():
             process_weights_after_loading(self.model_runner.model, self.model_config, self.device)
             # Also trigger drafter model's post processing if MTP is enabled
             if (
-                getattr(self.model_runner, "drafter", None) is not None
-                and getattr(self.model_runner.drafter, "model", None) is not None
+                hasattr(self.model_runner, "drafter")
+                and hasattr(self.model_runner.drafter, "model")
             ):
                 process_weights_after_loading(
                     self.model_runner.drafter.model, self.model_config, self.device