From 3d7cffc4a4d855676f66ed71adece0b68d26628c Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Mon, 24 Nov 2025 18:21:34 +0300
Subject: [PATCH] add noobai-rf to flow names

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

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 8a81aafc7..3e8aef5f8 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -657,7 +657,7 @@ def set_overrides(sd_model, checkpoint_info, model_type):
         # scheduler_config['beta_schedule'] = 'scaled_linear'
         # scheduler_config['timestep_spacing'] = 'trailing'
         sd_model.scheduler = diffusers.EulerAncestralDiscreteScheduler.from_config(scheduler_config)
-        if 'bigaspv25' in checkpoint_info_name or ('flow' in checkpoint_info_name and 'flower' not in checkpoint_info_name):
+        if 'bigaspv25' in checkpoint_info_name or 'noobai-rf' in checkpoint_info_name ('flow' in checkpoint_info_name and 'flower' not in checkpoint_info_name):
             scheduler_config = sd_model.scheduler.config
             scheduler_config['prediction_type'] = 'flow_prediction'
             scheduler_config['beta_schedule'] = 'linear'