TheLastBen / fast-stable-diffusion

fast-stable-diffusion + DreamBooth
MIT License
7.49k stars 1.31k forks source link

Something went wrong error while trying to start training #767

Open ReEnMikki opened 1 year ago

ReEnMikki commented 1 year ago
usage: train_dreambooth.py [-h] --pretrained_model_name_or_path
                           PRETRAINED_MODEL_NAME_OR_PATH [--revision REVISION]
                           [--tokenizer_name TOKENIZER_NAME]
                           --instance_data_dir INSTANCE_DATA_DIR
                           [--class_data_dir CLASS_DATA_DIR] --instance_prompt
                           INSTANCE_PROMPT [--class_prompt CLASS_PROMPT]
                           [--with_prior_preservation]
                           [--prior_loss_weight PRIOR_LOSS_WEIGHT]
                           [--num_class_images NUM_CLASS_IMAGES]
                           [--output_dir OUTPUT_DIR] [--seed SEED]
                           [--resolution RESOLUTION] [--center_crop]
                           [--train_text_encoder]
                           [--train_batch_size TRAIN_BATCH_SIZE]
                           [--sample_batch_size SAMPLE_BATCH_SIZE]
                           [--num_train_epochs NUM_TRAIN_EPOCHS]
                           [--max_train_steps MAX_TRAIN_STEPS]
                           [--save_steps SAVE_STEPS]
                           [--gradient_accumulation_steps GRADIENT_ACCUMULATION_STEPS]
                           [--gradient_checkpointing]
                           [--learning_rate LEARNING_RATE] [--scale_lr]
                           [--lr_scheduler LR_SCHEDULER]
                           [--lr_warmup_steps LR_WARMUP_STEPS]
                           [--use_8bit_adam] [--adam_beta1 ADAM_BETA1]
                           [--adam_beta2 ADAM_BETA2]
                           [--adam_weight_decay ADAM_WEIGHT_DECAY]
                           [--adam_epsilon ADAM_EPSILON]
                           [--max_grad_norm MAX_GRAD_NORM] [--push_to_hub]
                           [--hub_token HUB_TOKEN]
                           [--hub_model_id HUB_MODEL_ID]
                           [--logging_dir LOGGING_DIR]
                           [--mixed_precision {no,fp16,bf16}]
                           [--local_rank LOCAL_RANK]
train_dreambooth.py: error: unrecognized arguments: --image_captions_filename --save_starting_step=3000 --stop_text_encoder_training=10 --save_n_steps=3000 --Session_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/gtest
Traceback (most recent call last):
  File "/usr/local/bin/accelerate", line 8, in <module>
    sys.exit(main())
  File "/usr/local/lib/python3.7/dist-packages/accelerate/commands/accelerate_cli.py", line 43, in main
    args.func(args)
  File "/usr/local/lib/python3.7/dist-packages/accelerate/commands/launch.py", line 837, in launch_command
    simple_launcher(args)
  File "/usr/local/lib/python3.7/dist-packages/accelerate/commands/launch.py", line 354, in simple_launcher
    raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['/usr/bin/python3', '/content/diffusers/examples/dreambooth/train_dreambooth.py', '--image_captions_filename', '--save_starting_step=3000', '--stop_text_encoder_training=10', '--save_n_steps=3000', '--Session_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/gtest', '--pretrained_model_name_or_path=/content/stable-diffusion-v1-5', '--instance_data_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/gtest/instance_images', '--output_dir=/content/models/gtest', '--instance_prompt=', '--seed=682980', '--resolution=512', '--mixed_precision=fp16', '--train_batch_size=1', '--gradient_accumulation_steps=1', '--use_8bit_adam', '--learning_rate=2e-6', '--lr_scheduler=polynomial', '--lr_warmup_steps=0', '--max_train_steps=15000']' returned non-zero exit status 2.
Something went wrong

Error while running start training cell, please help

strikestorm0 commented 1 year ago

same here

brandnewx commented 1 year ago

TheLastBen reset his Diffusers repo 20 hours ago, and now this repo and all the collab books are no longer functioning.

TheLastBen commented 1 year ago

TheLastBen reset his Diffusers repo 20 hours ago, and now this repo and all the collab books are no longer functioning.

the current diffusers branch us "updt" not the "main" one, the reason of the temporary down is because google switched to python 3.8, but now everything is fixed

brandnewx commented 1 year ago

TheLastBen reset his Diffusers repo 20 hours ago, and now this repo and all the collab books are no longer functioning.

the current diffusers branch us "updt" not the "main" one, the reason of the temporary down is because google switched to python 3.8, but now everything is fixed

Thanks for everything, bro. You've done a great job! Your modifications to HuggingFace/Shivam Diffusers are amazing!

Mariiexe commented 1 year ago

Using python 3.8 but it still doesn't work for me ; statut 1 btw

TheLastBen commented 1 year ago

@Mariiexe use the latest colab and use the new method colab

Mariiexe commented 1 year ago

@TheLastBen It works now, thanks !