Training Issues!
Closed this issue · 2 comments
Henry-Bi commented
Thank you for your outstanding work on this project. I encountered an issue when following your training guide in step 4. When running the code, I encountered the following error: File "path/to/script.py", line 645, in from_pretrained raise ValueError( ValueError: Cannot load <class 'models.unet_2d_condition.UNet2DConditionModel'> from ./preset/models/stable-diffusion-2-base because the following keys are missing: .... Could you please advise on how to resolve this issue?
Henry-Bi commented
I have solved this problem by adding 'low_cpu_mem_usage=False' in train_seesr.py. More detialed discussion could be found in the above-mentioned link.