-
Notifications
You must be signed in to change notification settings - Fork 6.5k
[Dreambooth] Make compatible with alt diffusion #1470
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
[Dreambooth] Make compatible with alt diffusion #1470
Conversation
| subfolder="text_encoder", | ||
| revision=args.revision, | ||
| ) | ||
| model_class = text_encoder_config.architectures[0] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Since transformers 4.0 all configs have the parameter .architectures and they only have one member in the list
|
The documentation is not available anymore as the PR was closed or merged. |
patil-suraj
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me! Will try this out quickly and merge
|
|
||
| ### Using DreamBooth for other pipelines than Stable Diffusion | ||
|
|
||
| Altdiffusion also support dreambooth now, the runing comman is basically the same as abouve, all you need to do is replace the `MODEL_NAME` like this: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"runing", "comman", "abouve" are all misspelled fwiw
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks! feel free to open a PR to fix this :)
* [Dreambooth] Make compatible with alt diffusion * make style * add example
This PR makes the dreambooth training script more general by allowing all kinds of pipelines - e.g. it adds support for AltDiffusion and should be easier to maintain then: #1390