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

Fix TE1/TE2 learning rate handling on dreambooth SDXL #2697

Merged
merged 3 commits into from
Aug 9, 2024

Conversation

b-fission
Copy link
Contributor

@b-fission b-fission commented Aug 6, 2024

These changes allow the user to enable/disable training of the text encoders via the learning rates on SDXL dreambooth without having to manually specify command-line flags.

Fixed second part of issue #2542

bmaltais and others added 3 commits August 2, 2024 10:44
…eerror-int-argument-must-be-a-string-a-bytes-like-object-or-a-real-number-not-nonetype

Fix issue with missing key. Upgrade Gradio release for security issue.
SDXL dreambooth apparently trains without the text encoders by default,
requiring the `--train_text_encoder` flag to be passed so that the
learning rates for TE1/TE2 are recognized.

The toml handling now permits 0 to be passed as a learning rate in
order to disable training of one or both text encoders.
This behavior aligns with the description given on the GUI.

TE1/TE2 learning rate parameters can be left blank on the GUI to
not pass a value to the training script.
In python3, `0 == False` will evaluate True.
That can cause arg values of 0 to be wrongly eliminated from the toml output.
The conditional must check the type when comparing for False.
@FurkanGozukara
Copy link
Contributor

this error was fixed but returned back and broken my code

so in any case now i include in command line args :D

@bmaltais bmaltais changed the base branch from master to dev August 9, 2024 00:22
@bmaltais
Copy link
Owner

bmaltais commented Aug 9, 2024

Nice. Merging in dev. Thank you for taking the time to improve the GUI!

@bmaltais bmaltais merged commit 93b1c07 into bmaltais:dev Aug 9, 2024
1 check failed
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.

3 participants