Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Attempted fix: --resume_text_encoder in train_lora_dreambooth.py #114

Open
wants to merge 1 commit into
base: develop
Choose a base branch
from

Conversation

hdon96
Copy link
Contributor

@hdon96 hdon96 commented Jan 5, 2023

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)

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 <module>
    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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant