# FlowMatch Euler Discrete Scheduler for ComfyUI

9 steps, big res, zero noise.

**FlowMatchEulerDiscrete** seems not exposed in ComfyUI, but it is what the official Z-Image demo in diffusers use.

So:

- I am exposing it in the scheduler section for you to use within KSampler.
- On top I provide a node, experimental, to configure the scheduler for use with CustomSampler and play with.

In short...if you want **sharper and noise free images**, use this!

## Installation

- use comfy ui manager (search erosDiffusion or ComfyUI-EulerFlowMatchingDiscreteScheduler)

or

- `git clone https://github.com/erosDiffusion/ComfyUI-EulerDiscreteScheduler.git` in your custom nodes folder.

Example output (more below)
<img width="1792" height="1120" alt="example" src="https://github.com/user-attachments/assets/91d4f679-9c9e-40ef-bed6-12bb860c37e7" />

## What you get

- one new scheduler **FlowMatchEulerDiscreteScheduler** registered in the KSampler
- a custom node that exposes all parameters of the FlowMatchEulerDiscreteScheduler which Outputs **SIGMAS** for use with **SamplerCustom** node.

![highlight](https://github.com/user-attachments/assets/249cd15d-f373-46c7-bacb-13a4b5421ba3)

## Usage

- **Simple**: select the FlowMatchEulerDiscreteScheduler in the default workflow from ComfyUI and run.
- **Advanced/experimental**:
  1. Add **FlowMatch Euler Discrete Scheduler (Custom)** node to your workflow
  2. Connect its SIGMAS output to **SamplerCustom** node's sigmas input
  3. Adjust parameters to control the sampling behavior, you have ALL the parameters to play with.

## Troubleshoot
- if the scheduler does not appear when you have res4lyf package installed you can try:
  -- workaround 1: adding an samplerCustom node and connect the sigmas to a  basicScheduler node. this way the scheduler should be available in the list
  -- workaround 2: disable res4lyf if you don't need that
  -- workaround 3 use the flowmatch scheduler (custom) and connect to the sigmas of the samplerCustom.
- if your install fails you might have to use the correct version of peft package, some users reported this as issue, check startup logs and install the proper version

## Tech bits:

- https://huggingface.co/docs/diffusers/api/schedulers/flow_match_euler_discrete
- https://huggingface.co/Tongyi-MAI/Z-Image-Turbo/blob/main/scheduler/scheduler_config.json

## Find this useful and want to support ?

[Buy me a beer!](https://donate.stripe.com/cNi9ALaASf65clXahPcV201)

<img width="1920" height="1088" alt="ComfyUI_00716_" src="https://github.com/user-attachments/assets/bb2fc530-8a90-4180-96fb-adf6c48f5b09" />

More examples:
<img width="1536" height="1088" alt="image" src="https://github.com/user-attachments/assets/1ab561e7-b51d-413c-b788-13ed4fb6e129" />
<img width="1536" height="1088" alt="image" src="https://github.com/user-attachments/assets/1931af7e-1b3e-47c9-ac20-27add5135a71" />

## Changelog
**1.0.8**

- attempt fixing incompatibility with res4lyf by adding the scheduler to the list.
  
**1.0.7**

- nunchaku qwen patch fix, tiled diffusion patch fix
  users reported issues with dimensions not being handled correctly, this should fix it.


**1.0.6**

- updated example
- updated pyproject deps (diffusers)

**1.0.5**

- remove bad practice of forking diffusers install on error (requirements.txt and does not rollback your diffusers if available)

**1.0.4**

- add start and end step by Etupa, with some fixes (can be used for image to image or restart sampling)
  <img width="2880" height="960" alt="node_unknown" src="https://github.com/user-attachments/assets/247cb5ab-241f-43ce-b9d4-61c56ccb3711" />

**1.0.3**

- node publish action

**1.0.2**

- changed the device management in the custom scheduler node to be on gpu (cuda)
- removed flash attention node dependency from the custom scheduler node
- removed flash attention node from init
- added mit licensing