This repository is independently developed as a felxible framework to generate high-quality Image-Text pairs for finetuning Image-Generation models, such as Stable Diffusion, DALL-E, and other generative models. By leveraging open-source captioning models, GenIText automates the process of generating diverse captions for corresponding images, ensuring that the text data is well-suited for downstream applications such as style-specific generations or domain adaptation. This framework is designed to complement contemporary repositories or modules in the field, offering an additional option for flexibility and automation to create customized datasets.
GenIText will become distributable as a CLI tool once package is ready for testing across systems. Please support in any way you see fit!
GenIText is available as a Python package and can be installed easily using pip.
To install GenIText, simply run:
pip install genitextAfter installation, you can verify that the CLI tool is accessible by running:
genitext --helpTo initiate the CLI tool, run:
genitextGenIText incorporates LLMs from Ollama to assist with prompt refinement which means ollama has to be available on the device when running /refine in the CLI tool. You can download the software for Mac or Windows OS from here. For Linux OS, you can install directly via the following:
curl -fsSL https://ollama.com/install.sh | shAfter installing, pull the appropriate LLM you want to use in /refine. Currently, the default config is set to deepseek-r1:7b since it offers strong performance with it's reasoning capabilities while using relatively trivial memory. Options to switch the LLM in config will be made available soon.