diff --git a/src/lerobot/policies/pi05/modeling_pi05.py b/src/lerobot/policies/pi05/modeling_pi05.py index dd002d307..b017bbc57 100644 --- a/src/lerobot/policies/pi05/modeling_pi05.py +++ b/src/lerobot/policies/pi05/modeling_pi05.py @@ -1093,9 +1093,9 @@ class PI05Policy(PreTrainedPolicy): if self.config.rtc_config is not None: self.rtc_processor = RTCProcessor(self.config.rtc_config) - # Set rtc_processor to the model if it exists - if self.model is not None: - self.model.rtc_processor = self.rtc_processor + model_value = getattr(self, "model", None) + if model_value is not None: + model_value.rtc_processor = self.rtc_processor def _rtc_enabled(self) -> bool: return self.config.rtc_config is not None and self.config.rtc_config.enabled