Skip to content

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

@williamberman

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

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