Stops at first checkpoint

[maavangent]

This morning I created my first model. Great fun!

This afternoon I've tried to do a second experiment but now I get the following error when reaching the first checkpoint:

Progress:|████                     | 15% 450/3000 [07:30<41:51,  1.02it/s, loss=0.0615, lr=1.72e-6]  Freezing the text_encoder ... 
Traceback (most recent call last):
  File "/content/diffusers/examples/dreambooth/train_dreambooth.py", line 788, in <module>
    main()
  File "/content/diffusers/examples/dreambooth/train_dreambooth.py", line 714, in main
    text_encoder=accelerator.unwrap_model(text_encoder),
  File "/usr/local/lib/python3.7/dist-packages/diffusers/pipeline_utils.py", line 660, in from_pretrained
    f"Pipeline {pipeline_class} expected {expected_modules}, but only {passed_modules} were passed."
ValueError: Pipeline <class 'diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline'> expected {'safety_checker', 'tokenizer', 'feature_extractor', 'scheduler', 'vae', 'unet', 'text_encoder'}, but only {'tokenizer', 'scheduler', 'vae', 'unet', 'text_encoder'} were passed.
Progress:|████                     | 15% 450/3000 [07:31<42:37,  1.00s/it, loss=0.0615, lr=1.72e-6]
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', '--train_text_encoder', '--save_starting_step=500', '--stop_text_encoder_training=450', '--save_n_steps=500', '--Session_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/inkmvgnt', '--pretrained_model_name_or_path=/content/stable-diffusion-v1-5', '--instance_data_dir=/content/gdrive/MyDrive/Fast-Dreambooth/Sessions/inkmvgnt/instance_images', '--output_dir=/content/models/inkmvgnt', '--instance_prompt=', '--seed=902997', '--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=3000']' returned non-zero exit status 1.
Something went wrong

Any ideas how I can fix this?

[maavangent]

I'm using it in Google Colab by the way:
https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast-DreamBooth.ipynb

[TheLastBen]

I'll fix it

[maavangent]

Nice, thanks!

[TheLastBen]

this should fix it 52470d6

[maavangent]

O wow. That's quick!

I'll try it again.

[maavangent]

Yes, perfect. Works again. Nice one!

[TheLastBen]

Great !, make sure you update the colab regularly because of the v2 changes, sometimes things break