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

fixup! Add tests for flow matching models with RTC

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-11 22:30:57 +07:00
parent a59ebab66b
commit a14e8a65cd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi0/modeling_pi0.py
+1 -1
View File
@@ -916,8 +916,8 @@ class PI0Policy(PreTrainedPolicy):
self.config = config self.config = config
# Initialize the core PI0 model # Initialize the core PI0 model
self.init_rtc_processor()
self.model = PI0Pytorch(config, rtc_processor=self.rtc_processor) self.model = PI0Pytorch(config, rtc_processor=self.rtc_processor)
self.init_rtc_processor()
# Enable gradient checkpointing if requested # Enable gradient checkpointing if requested
if config.gradient_checkpointing: if config.gradient_checkpointing: