KeyError: 'time_embed.0.weight'

[thibaudart]

I launch a training, stop it and want to continue it.

I set the source model as the model of the latest epoch and the resume from saved training state to the latest state folder. Everyelse are similar.

I got this issue:

Traceback (most recent call last):
File "D:\kohya_ss\train_network.py", line 699, in
train(args)
File "D:\kohya_ss\train_network.py", line 126, in train
text_encoder, vae, unet, _ = train_util.load_target_model(args, weight_dtype)
File "D:\kohya_ss\library\train_util.py", line 2547, in load_target_model
text_encoder, vae, unet = model_util.load_models_from_stable_diffusion_checkpoint(args.v2, name_or_path)
File "D:\kohya_ss\library\model_util.py", line 877, in load_models_from_stable_diffusion_checkpoint
converted_unet_checkpoint = convert_ldm_unet_checkpoint(v2, state_dict, unet_config)
File "D:\kohya_ss\library\model_util.py", line 234, in convert_ldm_unet_checkpoint
new_checkpoint["time_embedding.linear_1.weight"] = unet_state_dict["time_embed.0.weight"]
KeyError: 'time_embed.0.weight'