search

Mikubill / naifu

Train generative models with pytorch lightning
MIT License
284 stars 36 forks source link

Conversion fails #14

Closed Dingo64 closed 1 year ago

Dingo64 commented 1 year ago

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'