Visit ComfyUI Online for ready-to-use ComfyUI environment
ComfyUI-TrainTools-MZ provides nodes for fine-tuning lora within ComfyUI, leveraging training tools like kohya-ss/sd-scripts to enhance customization and performance.
ComfyUI-TrainTools-MZ is an extension designed to facilitate the fine-tuning of LoRA (Low-Rank Adaptation) models within the ComfyUI environment. This extension leverages powerful training tools such as kohya-ss/sd-scripts to streamline the process of model training and customization. For AI artists, this means you can easily enhance and adapt your AI models to better suit your creative needs, without needing deep technical expertise.
At its core, ComfyUI-TrainTools-MZ provides a set of nodes that integrate with ComfyUI, allowing you to set up, configure, and execute training workflows for LoRA models. Think of these nodes as building blocks that you can connect to define how your model should be trained. Each node has specific functions, such as initializing the training environment, copying images, configuring training parameters, and running the training process. By connecting these nodes in a sequence, you can create a customized training pipeline that fits your specific requirements.
These nodes are used to initialize and train LoRA models using the HunYuanDiT branch of the sd-scripts. They set up the necessary workspace and execute the training process.
This node initializes the training folder within the output directory. You can specify:
This node copies images to the training folder and configures the dataset. Key options include:
This node provides basic training configurations. For detailed field references, see the .
This node offers more advanced training configurations. Again, refer to the for detailed field references.
This is the main training node. It includes options such as:
network_weights
parameter in sd-scripts.Here are some common issues and their solutions:
Solution: Ensure that the lora_name
, branch
, and source
parameters are correctly set in the MZ_KohyaSSInitWorkspace node. Verify that the output directory has the necessary write permissions.
Solution: Check the images
parameter in the MZ_ImagesCopyWorkspace node. Make sure the image paths are correct and that the source folder is accessible. If using the ComfyUI-VideoHelperSuite, ensure it is properly installed and configured.
Solution: Review the configurations in the MZ_KohyaSSUseConfig and MZ_KohyaSSAdvConfig nodes. Ensure all required fields are correctly filled. Refer to the for detailed configuration options.
For additional resources, tutorials, and community support, consider the following:
© Copyright 2024 RunComfy. All Rights Reserved.