diffusers
diffusers copied to clipboard
[modular] Tests for custom blocks in modular diffusers
What does this PR do?
As the title suggests, this PR adds a test suite for the custom blocks in modular diffusers.
TestModularCustomBlockstests for basic checks, e.g., properties, configs, components, inputs, etc. It also has a test that checks for Hub loading and its correctness.TestKreaCustomBlocksIntegrationchecks for correct loading and also the forward pass to ensure nothing is broken as we continue to improve modular diffusers.
Notes:
- I took the liberty to simplify the
.github/workflows/pr_modular_tests.yml, getting rid of the matrix as it wasn't really needed. - The tests were implemented in
pytestbecause we want to slowly move to it fromunittest.pytestmakes things easier to maintain long term. - The
TestKreaCustomBlocksIntegrationtest suite was checked in an H100 because it is a bit heavy.
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.