admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-20 16:31:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixup! Use output_dir for saving all evaluation images

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-03 19:07:01 +07:00
parent c409ed2d1d
commit bb23dafad1
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/smolvla/modeling_smolvla.py
+3 -1
View File
@@ -253,7 +253,9 @@ class SmolVLAPolicy(PreTrainedPolicy):
"""Initialize RTC processor if RTC is enabled in config.""" """Initialize RTC processor if RTC is enabled in config."""
self.rtc_processor = None 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) self.rtc_processor = RTCProcessor(self.config.rtc_config)
# In case of calling init_rtc_processor after the model is created # In case of calling init_rtc_processor after the model is created