Mikubill/naifu

Conversion fails

Dingo64 opened this issue · 0 comments

python convert_to_sd.py --src pruned/dreamshaper_332BakedVaeClipFix.ckpt --dst dream_sim.ckpt

it just shows errors:

Traceback (most recent call last):
  File "K:\3\convert_to_sd.py", line 345, in <module>
    unet_state_dict = convert_unet_state_dict(unet_state_dict, is_v2)
  File "K:\3\convert_to_sd.py", line 107, in convert_unet_state_dict
    new_state_dict = {v: unet_state_dict[k] for k, v in mapping.items()}
  File "K:\3\convert_to_sd.py", line 107, in <dictcomp>
    new_state_dict = {v: unet_state_dict[k] for k, v in mapping.items()}
KeyError: 'time_embedding.linear_1.weight'