-
Notifications
You must be signed in to change notification settings - Fork 6.6k
Closed
Closed
Copy link
Labels
bugSomething isn't workingSomething isn't working
Description
Describe the bug
I think there was a typo, because
no_flip not in args, only no_hflip
Reproduction
!accelerate launch train_diffusion_dpo_sdxl.py
--pretrained_model_name_or_path="/home/ubuntu/mydrive/auto/stable-diffusion-webui/models/Stable-diffusion/ModelStep3"
--pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix
--output_dir="Nebul-Redmond-SDXL"
--mixed_precision="fp16"
--dataset_name=kashif/pickascore
--train_batch_size=8
--gradient_accumulation_steps=1
--gradient_checkpointing
--use_8bit_adam
--rank=8
--learning_rate=8e-6
--report_to="wandb"
--lr_scheduler="constant"
--lr_warmup_steps=0
--max_train_steps=2000
--checkpointing_steps=500
--run_validation --validation_steps=50
--seed="0"
--report_to="wandb"
--push_to_hub
Logs
File "train_diffusion_dpo_sdxl.py", line 744, in preprocess_train
if not args.no_flip and random.random() < 0.5:
AttributeError: 'Namespace' object has no attribute 'no_flip'. Did you mean: 'no_hflip'?System Info
Platform: Linux
Python 3.10.13
torch==2.1.0
diffusers==0.25.1
transformers==4.35.2
accelerate==0.24.1
Using GPU in script?: yes
Who can help?
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working