NeuroImage

This is an image generation tool that implements the generating of images by the following diffusion models

• Kandinsky 2.2
• Kandinsky 3.0

Usage

git clone [email protected]:ixtal23/neuroimage.git

cd neuroimage

pip install -r requirements.txt

python main.py --output-file OUTPUT_FILE
               --prompt PROMPT
               [--negative-prompt NEGATIVE_PROMPT]
               [--seed SEED]
               [--guidance-scale GUIDANCE_SCALE]
               [--inference-steps [1-100]]
               [--height [64-1024]]
               [--width [64-1024]]
               [--kandinsky-version {2.2,3.0}]
               [-h]

Options

--output-file OUTPUT_FILE           a path to an output file
--prompt PROMPT                     prompt
--negative-prompt NEGATIVE_PROMPT   negative prompt
--seed SEED                         seed
--guidance-scale GUIDANCE_SCALE     guidance scale
--inference-steps [1-100]           the nummber of inference steps
--height [64-1024]                  the height in pixels of the generated image
--width [64-1024]                   the width in pixels of the generated image
--kandinsky-version {2.2,3.0}       the version of Kandinsky model
-h, --help                          show this help message and exit

Credits

Thanks a lot all developers behind libraries used in this project:

• Diffusers Community
• Diffusers GitHub
• Kandinsky Community
• Kandinsky 2.2 GitHub
• Kandinsky 3.0 GitHub