fix unipc-flowmatch

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-19 17:24:32 +02:00 · 2025-04-14 10:14:26 -04:00
parent 59efc95e00
commit 8f7a397a32
1 changed files with 2 additions and 0 deletions
@@ -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'