From e4cdcefdb7dfc42d7a033b5c5c6c5d1b8993320c Mon Sep 17 00:00:00 2001
From: blepping <blepping@tuta.io>
Date: Wed, 14 May 2025 13:46:42 -0600
Subject: [PATCH] Fix rectified flow type detection for non-Flux models

---
 py/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/py/model.py b/py/model.py
index 581c92b..e1b5894 100644
--- a/py/model.py
+++ b/py/model.py
@@ -167,7 +167,7 @@ class ModelCallCache:
         self.extra_args = extra_args
         self.cfg1_uncond_optimization = cfg1_uncond_optimization
         self.cfg_scale_override = cfg_scale_override
-        self.is_rectified_flow = x.shape[1] == 16 and isinstance(
+        self.is_rectified_flow = isinstance(
             model.inner_model.inner_model.model_sampling, comfy.model_sampling.CONST
         )
         self.latent_format = OCSLatentFormat(