Skip to content

Add additional integration testing around training scripts i.e. dreambooth #1917

Description

@williamberman

Is your feature request related to a problem? Please describe.
We have some integration tests for testing training scripts. These were added pre-dreambooth, so we could add some more!

Existing tests: https://github.com/huggingface/diffusers/blob/main/examples/test_examples.py
CI Call:

python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v --make-reports=examples_torch_cuda examples/

Describe the solution you'd like

  • Collect list of example scripts that could use more testing
  • Determine common flag combinations worth adding tests for
  • How e2e do we want to test? Is it worth doing full training runs and adding to nightly? -> we do cpu tests on merge

Status of tested scripts

  • train_dreambooth_flax.py
  • train_dreambooth_lora.py
  • train_dreambooth.py
  • train_text_to_image_flax.py
  • train_text_to_image_lora.py
  • train_text_to_image.py
  • textual_inversion_flax.py
  • textual_inversion.py
  • train_unconditional.py

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

staleIssues that haven't received updatestraining

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions