TypeError: All input tensors need to be on the same GPU, but found some tensors to not be on a GPU

Got this issue:
On cuda 12.1 ant pytorch 2.3.1 using LoRa
Setup: NVIDIA RTX 3050 Laptop GPU 4 GB VRAM and 16 GB RAM
ERROR lora diffusion_model.input_blocks.1.0.emb_layers.1.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.2.0.emb_layers.1.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.0.emb_layers.1.weight shape '[409600, 1]' is invalid for input of size 819200
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.4.1.transformer_blocks.1.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.0.emb_layers.1.weight shape '[409600, 1]' is invalid for input of size 819200
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.5.1.transformer_blocks.1.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.input_blocks.7.0.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.1.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.2.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.3.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.4.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.5.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.6.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.7.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.8.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.7.1.transformer_blocks.9.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.0.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.1.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.2.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.3.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.4.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.5.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.6.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.7.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.8.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.input_blocks.8.1.transformer_blocks.9.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.1.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.2.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.3.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.4.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.5.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.6.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.7.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.8.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.1.transformer_blocks.9.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.0.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.middle_block.2.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.0.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.0.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.1.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.2.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.3.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.4.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.5.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.6.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.7.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.8.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.0.1.transformer_blocks.9.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.0.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.0.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.1.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.2.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.3.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.4.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.5.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.6.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.7.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.8.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.1.1.transformer_blocks.9.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.0.emb_layers.1.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.0.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.1.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.2.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.3.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.4.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.5.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.6.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.7.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.8.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_out.0.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_v.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_q.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.2.1.transformer_blocks.9.attn1.to_k.weight shape '[819200, 1]' is invalid for input of size 1638400
ERROR lora diffusion_model.output_blocks.3.0.emb_layers.1.weight shape '[409600, 1]' is invalid for input of size 819200
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.3.1.transformer_blocks.1.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.0.emb_layers.1.weight shape '[409600, 1]' is invalid for input of size 819200
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.4.1.transformer_blocks.1.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.0.emb_layers.1.weight shape '[409600, 1]' is invalid for input of size 819200
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_out.0.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_v.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_q.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.5.1.transformer_blocks.1.attn1.to_k.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.6.0.emb_layers.1.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.7.0.emb_layers.1.weight shape '[204800, 1]' is invalid for input of size 409600
ERROR lora diffusion_model.output_blocks.8.0.emb_layers.1.weight shape '[204800, 1]' is invalid for input of size 409600
Moving model(s) has taken 1.33 seconds
0%| | 0/20 [00:00
samples = self.launch_sampling(steps, lambda: self.func(self.model_wrap_cfg, x, extra_args=self.sampler_extra_args, disable=False, callback=self.callback_state, **extra_params_kwargs))
File "D:\webui\system\python\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "D:\webui\webui\k_diffusion\sampling.py", line 594, in sample_dpmpp_2m
denoised = model(x, sigmas[i] * s_in, **extra_args)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "D:\webui\webui\modules\sd_samplers_cfg_denoiser.py", line 185, in forward
denoised, cond_pred, uncond_pred = sampling_function(self, denoiser_params=denoiser_params, cond_scale=cond_scale, cond_composition=cond_composition)
File "D:\webui\webui\backend\sampling\sampling_function.py", line 339, in sampling_function
denoised, cond_pred, uncond_pred = sampling_function_inner(model, x, timestep, uncond, cond, cond_scale, model_options, seed, return_full=True)
File "D:\webui\webui\backend\sampling\sampling_function.py", line 284, in sampling_function_inner
cond_pred, uncond_pred = calc_cond_uncond_batch(model, cond, uncond_, x, timestep, model_options)
File "D:\webui\webui\backend\sampling\sampling_function.py", line 254, in calc_cond_uncond_batch
output = model.apply_model(input_x, timestep_, **c).chunk(batch_chunks)
File "D:\webui\webui\backend\modules\k_model.py", line 43, in apply_model
model_output = self.diffusion_model(xc, t, context=context, control=control, transformer_options=transformer_options, **extra_conds).float()
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "D:\webui\webui\backend\nn\unet.py", line 713, in forward
h = module(h, emb, context, transformer_options)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "D:\webui\webui\backend\nn\unet.py", line 81, in forward
x = layer(x, emb, transformer_options)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "D:\webui\webui\backend\nn\unet.py", line 431, in forward
return checkpoint(self._forward, (x, emb, transformer_options), None, self.use_checkpoint)
File "D:\webui\webui\backend\nn\unet.py", line 12, in checkpoint
return f(*args)
File "D:\webui\webui\backend\nn\unet.py", line 454, in _forward
emb_out = self.emb_layers(emb).type(h.dtype)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\container.py", line 217, in forward
input = module(input)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\webui\system\python\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "D:\webui\webui\backend\operations.py", line 314, in forward
return functional_linear_4bits(x, self.weight, self.bias)
File "D:\webui\webui\backend\operations_bnb.py", line 10, in functional_linear_4bits
out = bnb.matmul_4bit(x, weight.t(), bias=bias, quant_state=weight.quant_state)
File "D:\webui\system\python\lib\site-packages\bitsandbytes\autograd\_functions.py", line 579, in matmul_4bit
return MatMul4Bit.apply(A, B, out, bias, quant_state)
File "D:\webui\system\python\lib\site-packages\torch\autograd\function.py", line 598, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
File "D:\webui\system\python\lib\site-packages\bitsandbytes\autograd\_functions.py", line 509, in forward
output = torch.nn.functional.linear(A, F.dequantize_4bit(B, quant_state).to(A.dtype).t(), bias)
File "D:\webui\system\python\lib\site-packages\bitsandbytes\functional.py", line 1358, in dequantize_4bit
is_on_gpu([A, absmax, out])
File "D:\webui\system\python\lib\site-packages\bitsandbytes\functional.py", line 432, in is_on_gpu
raise TypeError(
TypeError: All input tensors need to be on the same GPU, but found some tensors to not be on a GPU:
[(torch.Size([1, 204800]), device(type='cuda', index=0)), (torch.Size([6400]), device(type='cpu')), (torch.Size([320, 1280]), device(type='cuda', index=0))]
All input tensors need to be on the same GPU, but found some tensors to not be on a GPU:
[(torch.Size([1, 204800]), device(type='cuda', index=0)), (torch.Size([6400]), device(type='cpu')), (torch.Size([320, 1280]), device(type='cuda', index=0))]

No source files, tests, or entry points are named. Start by reproducing the failure with CUDA 12.1, PyTorch 2.3.1, LoRA, and the reported RTX 3050 setup, then trace the listed tensor-shape and GPU-placement errors; done means the triggering cause and a verified resolution are documented.

Written by the indexing model from the issue text.