Skip to content

realliyifei/llm-improve-diffusion

Repository files navigation

Improving Text-to-Image Diffusion Generation via Large Language Model

Environment

conda create -y -n diff python=3.9 cupy pkg-config compilers libjpeg-turbo opencv cudatoolkit=11.3 numba -c conda-forge
conda activate diff
pip install -r requirements.txt

Note: remember to fill in your OpenAI's API key in main.py

Imagine-Then-Verbalize

Execution example:

python main.py desc.txt -p template_prompt
python visualizer.py desc.txt -p template_prompt 

Results:

Sketch-Then-Draw

Execution example:

python visualize_sketch.py

Results: result folder and prompt folder

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published