From a14e8a65cdce9bee50f4d279c0371da539c8ccdd Mon Sep 17 00:00:00 2001
From: Eugene Mironov <helper2424@gmail.com>
Date: Tue, 11 Nov 2025 22:30:57 +0700
Subject: [PATCH] fixup! Add tests for flow matching models with RTC

---
 src/lerobot/policies/pi0/modeling_pi0.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lerobot/policies/pi0/modeling_pi0.py b/src/lerobot/policies/pi0/modeling_pi0.py
index f6e1d5f9c..c999e96c3 100644
--- a/src/lerobot/policies/pi0/modeling_pi0.py
+++ b/src/lerobot/policies/pi0/modeling_pi0.py
@@ -916,8 +916,8 @@ class PI0Policy(PreTrainedPolicy):
         self.config = config
 
         # Initialize the core PI0 model
-        self.init_rtc_processor()
         self.model = PI0Pytorch(config, rtc_processor=self.rtc_processor)
+        self.init_rtc_processor()
 
         # Enable gradient checkpointing if requested
         if config.gradient_checkpointing: