Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Dreambooth colab not working #2985

Open
Brandyfan opened this issue Jan 23, 2025 · 3 comments
Open

Dreambooth colab not working #2985

Brandyfan opened this issue Jan 23, 2025 · 3 comments

Comments

@Brandyfan
Copy link

I tried to resume training on one of my custom models and I keep getting an error when I use dreambooth. I tried deleting it from my google drive and did a fresh install before training a new model, but that didn't fix the problem. Any ideas?
This is the error I get-

Image
sys.exit(main())
^^^^^^
File "/usr/local/lib/python3.11/dist-packages/accelerate/commands/accelerate_cli.py", line 43, in main
args.func(args)
File "/usr/local/lib/python3.11/dist-packages/accelerate/commands/launch.py", line 837, in launch_command
simple_launcher(args)
File "/usr/local/lib/python3.11/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', '--train_only_unet', '--save_starting_step=2500', '--save_n_steps=210', '--Session_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/Isa', '--pretrained_model_name_or_path=/content/stable-diffusion-custom', '--instance_data_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/Isa/instance_images', '--output_dir=/content/models/Isa', '--captions_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/Isa/captions', '--instance_prompt=', '--seed=292560', '--resolution=1024', '--mixed_precision=fp16', '--train_batch_size=1', '--gradient_accumulation_steps=1', '--gradient_checkpointing', '--use_8bit_adam', '--learning_rate=2e-06', '--lr_scheduler=linear', '--lr_warmup_steps=0', '--max_train_steps=3510']' returned non-zero exit status 1.
Something went wrong

@Brandyfan
Copy link
Author

I just tried resuming the training on a different account and now it's working. I'm not sure why it's working on one and not the other. Hmmm

@Brandyfan
Copy link
Author

And now my other account gets the same error. Anyone else getting this error?

@Brandyfan
Copy link
Author

I did some research and found a workaround. Adding this code makes it work for now-

!apt-get update && apt-get install -y python3.10
!apt --fix-broken install

I still get the error, but Dreambooth trains just fine.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant