Bowen Bao c7617e482a Register BaseOutput subclasses as supported torch.utils._pytree nodes (#5459) ...

* Register BaseOutput subclasses as supported torch.utils._pytree nodes

* lint

---------

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

2023-10-24 15:01:47 +05:30

..

test_check_copies.py

[docs] Clean scheduler api (#4204)

2023-08-09 09:00:35 -07:00

test_check_dummies.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_config.py

keep _use_default_values as a list type (#4040)

2023-07-11 18:21:08 +02:00

test_dependencies.py

[Tests] add test for pipeline import. (#4276)

2023-07-28 00:08:15 +05:30

test_ema.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_hub_utils.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_image_processor.py

Fix type annotation (#5146)

2023-09-25 19:26:39 +02:00

test_outputs.py

Register BaseOutput subclasses as supported torch.utils._pytree nodes (#5459)

2023-10-24 15:01:47 +05:30

test_training.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_utils.py

[Pipeline utils] feat: implement push_to_hub for standalone models, schedulers as well as pipelines (#4128)

2023-08-15 07:39:22 +05:30