From 8f7a397a3251f17590ea84f02a080a06c9c0f03c Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Mon, 14 Apr 2025 10:14:26 -0400
Subject: [PATCH] fix unipc-flowmatch

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 modules/sd_samplers_diffusers.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/sd_samplers_diffusers.py b/modules/sd_samplers_diffusers.py
index fd58e2f28..97a6a7b26 100644
--- a/modules/sd_samplers_diffusers.py
+++ b/modules/sd_samplers_diffusers.py
@@ -284,6 +284,8 @@ class DiffusionSampler:
             del self.config['beta_end']
             del self.config['beta_schedule']
             del self.config['prediction_type']
+        if 'prediction_type' in self.config and 'Flow' in name:
+            self.config['prediction_type'] = 'flow_prediction'
         if 'SGM' in name:
             self.config['timestep_spacing'] = 'trailing'