modelscope / DiffSynth-Studio

Enjoy the magic of Diffusion models!
Apache License 2.0
6.4k stars 575 forks source link

where is the output_path of lora train? #122

Closed lonngxiang closed 2 months ago

lonngxiang commented 2 months ago
CUDA_VISIBLE_DEVICES=0 python examples/train/kolors/train_kolors_lora.py   --pretrained_unet_path models/kolors/Kolors/unet/diffusion_pytorch_model.safetensors   --pretrained_text_encoder_path models/kolors/Kolors/text_encoder   --pretrained_fp16_vae_path models/sdxl-vae-fp16-fix/diffusion_pytorch_model.safetensors   --dataset_path data/dog   --output_path ./models   --max_epochs 10   --center_crop   --use_gradient_checkpointing   --precision "16-mixed"

output_path ./models/lora ,The trained lora model is not seen in the folder image image

lonngxiang commented 2 months ago

找到,在lightning_logs下面