Skip to content

Latest commit

 

History

History

depth_canny

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
assets
assets
 
 
README.md
README.md
 
 
run_gradio.py
run_gradio.py
 
 
utils.py
utils.py
 
 
vars.py
vars.py
 
 

README.md

Nunchaku INT4 FLUX.1 Depth/Canny-to-Image Demo

demo

This interactive Gradio application transforms your uploaded image into a different style based on a text prompt. The generated image preserves either the depth map or Canny edge of the original image, depending on the selected model.

The base models are:

First you need to install some dependencies:

pip install git+https://github.com/asomoza/image_gen_aux.git
pip install controlnet_aux mediapipe

Then run:

python run_gradio.py
  • By default, the model is FLUX.1-Depth-dev. You can add -m canny to switch to FLUX.1-Canny-dev.
  • The demo loads the Gemma-2B model as a safety checker by default. To disable this feature, use --no-safety-checker.
  • To further reduce GPU memory usage, you can enable the W4A16 text encoder by specifying --use-qencoder.
  • By default, we use our INT4 model. Use -p bf16 to switch to the BF16 model.