[cm] Changing prepare for inference

Neutone · Feb 1, 2024 · 8c1b11a · 8c1b11a
1 parent 5d0e1f1
commit 8c1b11a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neutone_sdk/conv.py b/neutone_sdk/conv.py
@@ -268,8 +268,8 @@ def prepare_for_inference(self) -> None:
             log.info(f"Converting Conv1dGeneral to cached in prepare_for_inference()")
             self.set_cached(True)
         self.debug_mode = False
-        self.conv1d.eval()
         self.padding_cached.prepare_for_inference()
+        self.eval()
 
     def forward(self, x: Tensor) -> Tensor:
         """