Error when i run kandinsky2.2 #7279
leedaehan-kev
started this conversation in
General
Replies: 1 comment
-
I suspect you're passing wrong |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I confirmed that the prior code runs without issues. However, when executing the decoder code, this error is encountered:
This is the command I used:
#I have confirmed that the dataset name is set correctly.
accelerate launch --mixed_precision="fp16" train_text_to_image_decoder.py --dataset_name=$DATASET_NAME --resolution=768 --train_batch_size=1 --gradient_accumulation_steps=4 --gradient_checkpointing --max_train_steps=15000 --learning_rate=1e-05 --max_grad_norm=1 --checkpoints_total_limit=3 --lr_scheduler="constant" --lr_warmup_steps=0 --validation_prompts="A robot pokemon, 4k photo" --report_to="wandb" --push_to_hub --output_dir="kandi2-decoder-pokemon-model"
Thank you.
Beta Was this translation helpful? Give feedback.
All reactions