2025-01-09 19:40:13 +08:00
2025-01-09 19:40:13 +08:00
2025-01-09 19:40:13 +08:00
2025-01-07 17:06:20 +08:00
2025-01-08 14:32:48 +08:00
2025-01-09 19:40:13 +08:00
2025-01-09 19:40:13 +08:00

ComfyUI-TeaCache

Introduction

Timestep Embedding Aware Cache (TeaCache) is a training-free caching approach that estimates and leverages the fluctuating differences among model outputs across timesteps, thereby accelerating the inference. TeaCache works well for Image Diffusion models, Video Diffusion Models, and Audio Diffusion Models.

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.

Updates

  • Jan 9 2025: ComfyUI-TeaCache supports HunyuanVideo:
    • It can achieve a 1.6x lossless speedup and a 2x speedup without much visual quality degradation, which are consistent with the original TeaCache4HunyuanVideo.
  • Jan 8 2025: ComfyUI-TeaCache supports FLUX:
    • It can achieve a 1.4x lossless speedup and a 2x speedup without much visual quality degradation, which are consistent with the original TeaCache4FLUX.
    • Support FLUX LoRA!
    • Support FLUX ControlNet!

Installation

  1. Go to comfyUI custom_nodes folder, ComfyUI/custom_nodes/
  2. git clone https://github.com/welltop-cn/ComfyUI-TeaCache.git

Usage

The demo workflow is placed in examples folder.

Demo

  • FLUX

https://github.com/user-attachments/assets/e977cf34-f7d0-4b25-a2e3-10fd62ebfe30

  • HunyuanVideo

https://github.com/user-attachments/assets/4d8e9f12-2c54-40c5-a992-c2cecbde019a

Result comparison

  • FLUX

  • HunyuanVideo

https://github.com/user-attachments/assets/b3aca64d-c2ae-440c-a362-f3a7b6c633e0

Roadmap

  • Support LTX-Video

Acknowledgments

Thanks to TeaCache repo owner ali-vilab/TeaCache: Timestep Embedding Tells: It's Time to Cache for Video Diffusion Model

S
Description
No description provided
Readme Apache-2.0
44 MiB
Languages
Python 100%