ComfyUI InvSR

This project is an unofficial ComfyUI implementation of InvSR (Arbitrary-steps Image Super-resolution via Diffusion Inversion)

Installation

Navigate to the ComfyUI /custom_nodes directory

git clone https://github.com/yuvraj108c/ComfyUI_InvSR
cd ComfyUI_InvSR

# Requirements
pip install -e ".[torch]"
pip install -r requirements.txt

Usage

• Load example workflow
• Diffusers model (stabilityai/sd-turbo) will download automatically to ComfyUI/models/diffusers
• InvSR model (noise_predictor_sd_turbo_v5.pth) will download automatically to ComfyUI_InvSR/weights
• To deal with large images, e.g, 1k---->4k, set chopping_size 256
• If your GPU memory is limited, please set chopping_batch_size to 1

Citation

@article{yue2024InvSR,
  title={Arbitrary-steps Image Super-resolution via Diffusion Inversion},
  author={Yue, Zongsheng and Kang, Liao and Loy, Chen Change},
  journal = {arXiv preprint arXiv:2412.09013},
  year={2024},
}

License

This project is licensed under NTU S-Lab License 1.0