Hi,
I am trying to load lllyasviel/sd-controlnet-canny · Hugging Face as a checkpoint for train_controlnet.py
But I don’t have the required files (scheduler.bin, optimizer.bin, random_states_0.pkl)
I tried putting these files from a checkpoint of my own but I get the following error
Resuming from checkpoint controlnet
07/15/2024 04:51:30 - INFO - accelerate.accelerator - Loading states from train_controlnet_test_custom_dataset/controlnet
07/15/2024 04:51:33 - INFO - accelerate.checkpointing - All model weights loaded successfully
07/15/2024 04:51:38 - INFO - accelerate.checkpointing - All optimizer states loaded successfully
07/15/2024 04:51:38 - INFO - accelerate.checkpointing - All scheduler states loaded successfully
07/15/2024 04:51:38 - INFO - accelerate.checkpointing - All dataloader sampler states loaded successfully
07/15/2024 04:51:38 - INFO - accelerate.checkpointing - Could not load random states
Traceback (most recent call last):
File "/content/drive/MyDrive/train_controlnet.py", line 1187, in <module>
main(args)
File "/content/drive/MyDrive/train_controlnet.py", line 1018, in main
accelerator.load_state(os.path.join(args.output_dir, path))
File "/usr/local/lib/python3.10/dist-packages/accelerate/accelerator.py", line 3147, in load_state
self.step = override_attributes["step"]
KeyError: 'step'
Please Help