目录

MaskDiffusion: Boosting Text-to-Image Consistency with Conditional Mask (IJCV)

🚩 TODO/Updates

  • Basic Code.
  • Demo
  • Integration with LLM Priors.
  • Support for Other Pre-trained Models

Setup

Environment

Our code builds on the requirement of the official Stable Diffusion repository. To set up their environment, please run:

conda env create -f environment/environment.yaml
conda activate maskdiffusion

Testing

Testing with the mini-testset

python run_maskdiffusion.py

Acknowledgements

This orignal code is builds on the code from the diffusers library and Prompt-to-Prompt. When reorganizing the code to be public, I also reused code from Attend-and-Excite and Densediffusion, to achieve a more concise implementation.

关于
42.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

版权所有:中国计算机学会技术支持:开源发展技术委员会
京ICP备13000930号-9 京公网安备 11010802032778号