#160 · Primary category: Image Generation
Lora-Training-in-Comfy
This custom node lets you train LoRA directly in ComfyUI!
Project last updated:08/05/24
GitHub Stars
546
Forks
78
Contributors
2
License
Other
Why we included this project
Training a LoRA usually means leaving ComfyUI for a separate script, but this custom node keeps the whole loop inside the node graph. You give it a folder of captioned images, pick a name, set a few hyperparameters, and it writes the finished LoRA straight into ComfyUI's lora folder, so a refresh is all it takes to load and test it. The Advanced node adds more training options, and the Tensorboard node gives you a URL to watch loss logs while a run is going. It works with LCM, SD 2.0, and SD Turbo models, though the author hasn't tested SDXL. This is a community project, so some features are rough around the edges and you may need to tweak values by hand, but for artists who already live in ComfyUI it shortens the distance between training and testing considerably.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
stable-diffusion-webui
Stable Diffusion web UI
ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
upscayl
🆙 Upscayl - #1 Free and Open Source AI Image Upscaler for Linux, MacOS and Windows.
InvokeAI
InvokeAI is a leading open-source creative engine for Stable Diffusion, offering an industry-leading web UI for generating and editing visual media.