lora icon indicating copy to clipboard operation
lora copied to clipboard

Attempted fix: --resume_text_encoder in train_lora_dreambooth.py

Open hdon96 opened this issue 2 years ago • 0 comments

This attempted fix results in the following error. Putting it here in case someone can see the problem with inject_trainable_lora. Any help appreciated, thanks!

Traceback (most recent call last): File "C:\Users<user>\lora\train_lora_dreambooth.py", line 1044, in main(args) File "C:\Users<user>\lora\train_lora_dreambooth.py", line 631, in main unet_lora_params, _ = inject_trainable_lora( File "C:\Users<user>\lora\lora_diffusion\lora.py", line 176, in inject_trainable_lora _module._modules[name].lora_up.weight = loras.pop(0) File "F:\ANACONDA\envs\sd\lib\site-packages\torch\nn\modules\module.py", line 1228, in setattr raise TypeError("cannot assign '{}' as parameter '{}' " TypeError: cannot assign 'torch.HalfTensor' as parameter 'weight' (torch.nn.Parameter or None expected)

hdon96 avatar Jan 05 '23 15:01 hdon96