Skip to content

Conversation

@patrickvonplaten
Copy link
Contributor

@patrickvonplaten patrickvonplaten commented Nov 29, 2022

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

subfolder="text_encoder",
revision=args.revision,
)
model_class = text_encoder_config.architectures[0]
Copy link
Contributor Author

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

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Nov 29, 2022

The documentation is not available anymore as the PR was closed or merged.

Copy link
Contributor

@patil-suraj patil-suraj left a 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

@patil-suraj patil-suraj merged commit 1d4ad34 into main Nov 30, 2022
@patil-suraj patil-suraj deleted the make_dreambooth_compatible_with_alt_diffusion branch November 30, 2022 12:48

### 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:

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

Copy link
Contributor

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 :)

sliard pushed a commit to sliard/diffusers that referenced this pull request Dec 21, 2022
* [Dreambooth] Make compatible with alt diffusion

* make style

* add example
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.

5 participants