Skip to content

Commit 4bca908

Browse files
committed
Support HiDream-I1-Fast
1 parent bf45cdf commit 4bca908

5 files changed

Lines changed: 637 additions & 8 deletions

File tree

README.md

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,9 @@ Timestep Embedding Aware Cache ([TeaCache](https://github.com/ali-vilab/TeaCache
66
TeaCache has now been integrated into ComfyUI and is compatible with the ComfyUI native nodes. ComfyUI-TeaCache is easy to use, simply connect the TeaCache node with the ComfyUI native nodes for seamless usage.
77

88
## Updates
9+
- Jun 22 2025: ComfyUI-TeaCache supports HiDream-I1-Fast:
10+
- It can achieve a 1.4x lossless speedup and a 1.7x speedup without much visual quality degradation for HiDream-I1-Fast.
11+
- Support HiDream-I1-Fast LoRA!
912
- Jun 15 2025: ComfyUI-TeaCache supports HiDream-I1-Dev and Lumina-Image-2.0, adds cache_device option:
1013
- It can achieve a 1.5x lossless speedup and a 2x speedup without much visual quality degradation for HiDream-I1-Dev.
1114
- Support HiDream-I1-Dev LoRA!
@@ -68,8 +71,9 @@ To use TeaCache node, simply add `TeaCache` node to your workflow after `Load Di
6871
|:----------------------------:|:-----------------:|:-----------------:|:-----------------:|:-----------------:|
6972
| FLUX | 0.4 | 0 | 1 | ~2x |
7073
| PuLID-FLUX | 0.4 | 0 | 1 | ~1.7x |
71-
| HiDream-I1-Dev | 1 | 0 | 1 | ~2x |
7274
| HiDream-I1-Full | 0.35 | 0.1 | 1 | ~2x |
75+
| HiDream-I1-Dev | 1 | 0 | 1 | ~2x |
76+
| HiDream-I1-Fast | 1.5 | 0 | 0.9 | ~1.7x |
7377
| Lumina-Image-2.0 | 0.38 | 0.2 | 1 | ~1.7x |
7478
| HunyuanVideo | 0.15 | 0 | 1 | ~1.9x |
7579
| LTX-Video | 0.06 | 0 | 1 | ~1.7x |
@@ -89,7 +93,7 @@ If the image/video after applying TeaCache is of low quality, please reduce rel_
8993

9094
If you have enough VRAM, please select `cuda` in the `cache_device` option, which can bring faster inference, but will increase VRAM slightly. If you have limited VRAM, please select `cpu` in the `cache_device` option, which do not increase VRAM, but will make inference slower slightly.
9195

92-
The demo workflows ([flux](./examples/flux.json), [pulid_flux](./examples/pulid_flux.json), [hidream_i1_dev](./examples/hidream_i1_dev.json), [hidream_i1_full](./examples/hidream_i1_full.json), [lumina_image_2](./examples/lumina_image_2.json), [hunyuanvideo](./examples/hunyuanvideo.json), [ltx_video](./examples/ltx_video.json), [cogvideox](./examples/cogvideox.json), [wan2.1_t2v](./examples/wan2.1_t2v.json) and [wan2.1_i2v](./examples/wan2.1_i2v.json)) are placed in examples folder.
96+
The demo workflows ([flux](./examples/flux.json), [pulid_flux](./examples/pulid_flux.json), [hidream_i1_full](./examples/hidream_i1_full.json), [hidream_i1_dev](./examples/hidream_i1_dev.json), [hidream_i1_fast](./examples/hidream_i1_fast.json), [lumina_image_2](./examples/lumina_image_2.json), [hunyuanvideo](./examples/hunyuanvideo.json), [ltx_video](./examples/ltx_video.json), [cogvideox](./examples/cogvideox.json), [wan2.1_t2v](./examples/wan2.1_t2v.json) and [wan2.1_i2v](./examples/wan2.1_i2v.json)) are placed in examples folder.
9397

9498
### Compile Model
9599
To use Compile Model node, simply add `Compile Model` node to your workflow after `Load Diffusion Model` node or `TeaCache` node. Compile Model uses `torch.compile` to enhance the model performance by compiling model into more efficient intermediate representations (IRs). This compilation process leverages backend compilers to generate optimized code, which can significantly speed up inference. The compilation may take long time when you run the workflow at first, but once it is compiled, inference is extremely fast. The usage is shown below:
@@ -102,11 +106,14 @@ To use Compile Model node, simply add `Compile Model` node to your workflow afte
102106
- <p><strong>PuLID-FLUX</strong></p>
103107
![](./assets/compare_pulid_flux.png)
104108

109+
- <p><strong>HiDream-I1-Full</strong></p>
110+
![](./assets/compare_hidream_i1_full.png)
111+
105112
- <p><strong>HiDream-I1-Dev</strong></p>
106113
![](./assets/compare_hidream_i1_dev.png)
107114

108-
- <p><strong>HiDream-I1-Full</strong></p>
109-
![](./assets/compare_hidream_i1_full.png)
115+
- <p><strong>HiDream-I1-Fast</strong></p>
116+
![](./assets/compare_hidream_i1_fast.png)
110117

111118
- <p><strong>Lumina-Image-2.0</strong></p>
112119
![](./assets/compare_lumina_image_2.png)

assets/compare_hidream_i1_fast.png

932 KB
Loading

0 commit comments

Comments
 (0)