Kohya_ssでLoRA学習させようとしてエラーを吐く

Kohya_ssを用いてLoRAの学習をさせようとしていたら次のようなエラーを吐いた。

File "■\module.py", line 1671, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel:
        size mismatch for down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k.weight: copying a param with shape torch.Size([320, 768]) from checkpoint, the shape in current model is torch.Size([320, 1024]).
 ～
Traceback (most recent call last):
  File "■\runpy.py", line 196, in _run_module_as_main
～
    raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
～

結論はソースモデルにv2のチェックが入っていたことだった。えぇ…

ちなみに、たぶんこれを見に来た人は次でまた止まると思うので…

note.com

このページを参考に…