create LoRA network. base dim (rank): 64, alpha: 32
neuron dropout: p=None, rank dropout: p=None, module dropout: p=None
create LoRA for Text Encoder: 72 modules.
create LoRA for U-Net: 192 modules.
enable LoRA for text encoder
enable LoRA for U-Net
Traceback (most recent call last):
File "D:\lora_lian\sd-scripts\train_network.py", line 873, in <module>
train(args)
File "D:\lora_lian\sd-scripts\train_network.py", line 242, in train
info = network.load_weights(args.network_weights)
File "D:\lora_lian\sd-scripts\networks\lora.py", line 884, in load_weights
info = self.load_state_dict(weights_sd, False)
File "D:\lora_lian\python\lib\site-packages\torch\nn\modules\module.py", line 2041, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for LoRANetwork:
size mismatch for lora_unet_mid_block_attentions_0_proj_out.lora_up.weight: copying a param with shape torch.Size([1280, 128, 1, 1]) from checkpoint, the shape in current model is torch.Size([1280, 64, 1, 1]).
Traceback (most recent call last):
File "D:\lora_lian\python\lib\runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "D:\lora_lian\python\lib\runpy.py", line 86, in _run_code
exec(code, run_globals)
File "D:\lora_lian\python\lib\site-packages\accelerate\commands\launch.py", line 1114, in <module>
main()
File "D:\lora_lian\python\lib\site-packages\accelerate\commands\launch.py", line 1110, in main
launch_command(args)
File "D:\lora_lian\python\lib\site-packages\accelerate\commands\launch.py", line 1104, in launch_command
simple_launcher(args)
File "D:\lora_lian\python\lib\site-packages\accelerate\commands\launch.py", line 567, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['D:\\lora_lian\\python\\python.exe', './sd-scripts/train_network.py', '--config_file', 'D:\\lora_lian\\toml\\autosave\\20230709-112914.toml']' returned non-zero exit status 1.
Training failed / 训练失败