From bb23dafad1fec93e9efa4aa3aa9f4dbed4053139 Mon Sep 17 00:00:00 2001 From: Eugene Mironov Date: Mon, 3 Nov 2025 19:07:01 +0700 Subject: [PATCH] fixup! Use output_dir for saving all evaluation images --- src/lerobot/policies/smolvla/modeling_smolvla.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/lerobot/policies/smolvla/modeling_smolvla.py b/src/lerobot/policies/smolvla/modeling_smolvla.py index 02ec09421..98158db67 100644 --- a/src/lerobot/policies/smolvla/modeling_smolvla.py +++ b/src/lerobot/policies/smolvla/modeling_smolvla.py @@ -253,7 +253,9 @@ class SmolVLAPolicy(PreTrainedPolicy): """Initialize RTC processor if RTC is enabled in config.""" self.rtc_processor = None - if self.config.rtc_config is not None and self.config.rtc_config.enabled: + # Lets create processor if the config provided + # If RTC is not enabled - we still can track the denoising data + if self.config.rtc_config is not None: self.rtc_processor = RTCProcessor(self.config.rtc_config) # In case of calling init_rtc_processor after the model is created