Closed lonngxiang closed 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
找到,在lightning_logs下面
output_path ./models/lora ,The trained lora model is not seen in the folder