-
Notifications
You must be signed in to change notification settings - Fork 6.1k
Closed
Labels
Description
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:
diffusers/.github/workflows/push_tests.yml
Line 145 in 19a0ce4
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