LarryJane491 / Lora-Training-in-Comfy

This custom node lets you train LoRA directly in ComfyUI!
385 stars 56 forks source link

Train finished Prompt executed in 1.20 seconds #79

Open fjm-aw opened 1 month ago

fjm-aw commented 1 month ago

What could be the problem?

got prompt
[]
D:\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\Lora-Training-in-Comfy/sd-scripts/train_network.py
python -m accelerate.commands.launch --num_cpu_threads_per_process=8 "D:/ComfyUI/ComfyUI_windows_portable/ComfyUI/custom_nodes/Lora-Training-in-Comfy/sd-scripts/train_network.py" --enable_bucket --pretrained_model_name_or_path=D:/Models/models/checkpoints\DreamShaper_8_pruned.safetensors --train_data_dir="C:/Users/User/Desktop/TrainData" --output_dir="D:\Models\models\loras" --logging_dir="./logs" --log_prefix=super_20241018 --resolution=512,512 --network_module=networks.lora --max_train_epochs=10 --learning_rate=1e-4 --unet_lr=1e-4 --text_encoder_lr=1e-5 --lr_scheduler=cosine_with_restarts --lr_warmup_steps=0 --lr_scheduler_num_cycles=1 --network_dim=32 --network_alpha=32 --output_name=super_20241018 --train_batch_size=1 --save_every_n_epochs=100 --mixed_precision="fp16" --save_precision="fp16" --seed=22 --cache_latents --prior_loss_weight=1 --max_token_length=225 --caption_extension=".txt" --save_model_as=safetensors --min_bucket_reso=256 --max_bucket_reso=1584 --keep_tokens=0 --xformers --shuffle_caption --clip_skip=2 --optimizer_type=AdamW8bit --persistent_data_loader_workers --log_with=tensorboard --clip_skip=2 --optimizer_type=AdamW8bit --persistent_data_loader_workers --log_with=tensorboard
Train finished
Prompt executed in 1.20 seconds
Frontesque commented 2 days ago

I was getting this issue when all of the dependencies were installed. make sure you cd to the module folder and run pip install -r requirements.txt, oh and you need to make sure that you have the rustlang installed.