fixup! Use output_dir for saving all evaluation images

This commit is contained in:
Eugene Mironov
2025-11-03 19:07:01 +07:00
parent c409ed2d1d
commit bb23dafad1

View File

@@ -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