d8ahazard / sd_dreambooth_extension

Other
1.85k stars 281 forks source link

[Bug]: train completed lora model not work #1380

Closed yuan6785 closed 8 months ago

yuan6785 commented 8 months ago

Is there an existing issue for this?

What happened?

ubuntu 22.04

sd web ui v16.0 sd_dreambooth_extension: d8e01e1fd13dd4b6afadbb0382875f905ee5f429 sd model: v1-5-pruned-emaonly.safetensors [6ce0161689] dreambooth select model type: v1.x select: save lora extra weights

when i use lora mode in sd-web-ui tab "lora"

sd web ui report logs, lora not work:

[2023-11-03 13:41:02,634][DEBUG][root] - Network /mnt/sdwebui_public/versions/sdwebui_env/stable-diffusion-webui-20231102/stable-diffusion-webui/models/Lora/yxtest5_200.safetensors didn't match keys: {'lora_unet_input_blocks_1_0_transformer_blocks_0_attn1_to_k.alpha': 'diffusion_model_input_blocks_1_0_transformer_blocks_0_attn1_to_k', 'lora_unet_input_blocks_1_0_transformer_blocks_0_attn1_to_k.lora_down.weight': 'diffusion_model_input_blocks_1_0_transformer_blocks_0_attn1_to_k', .....}

i try without or use https://github.com/KohakuBlueleaf/a1111-sd-webui-lycoris.git
that's same error;

Steps to reproduce the problem

No

No

Command Line Arguments

***** Running training *****
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Num batches each epoch = 10
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Num Epochs = 20
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Batch Size Per Device = 1
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Gradient Accumulation steps = 1
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Total train batch size (w. parallel, distributed & accumulation) = 1
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Text Encoder Epochs: 20
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Total optimization steps = 200
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Total training steps = 200
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   Resuming from checkpoint: False
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   First resume epoch: 0
[2023-11-03 13:43:40,305][DEBUG][dreambooth.train_dreambooth] -   First resume step: 0
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   Lora: True, Optimizer: 8bit AdamW, Prec: fp16
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   Gradient Checkpointing: True
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   EMA: True
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   UNET: True
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   Freeze CLIP Normalization Layers: False
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   LR (Lora): 0.0001
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   Tenc LR (Lora): 5e-05
[2023-11-03 13:43:40,306][DEBUG][dreambooth.train_dreambooth] -   V2: False

Console logs

No

Additional information

No

github-actions[bot] commented 8 months ago

This issue is stale because it has been open 5 days with no activity. Remove stale label or comment or this will be closed in 5 days