diffusers/tests
Patrick von Platen cc59b05635
[ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334)
* add outputs for models

* add for pipelines

* finish schedulers

* better naming

* adapt tests as well

* replace dict access with . access

* make schedulers works

* finish

* correct readme

* make  bcp compatible

* up

* small fix

* finish

* more fixes

* more fixes

* Apply suggestions from code review

Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update src/diffusers/models/vae.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Adapt model outputs

* Apply more suggestions

* finish examples

* correct

Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-09-05 14:49:26 +02:00
..
__init__.py fix issues with loading, add test for pipeline 2022-06-07 15:40:36 +02:00
test_config.py [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_layers_utils.py Revive `make quality` (#203) 2022-08-17 15:22:04 +02:00
test_modeling_common.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00
test_models_unet.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00
test_models_vae.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00
test_models_vq.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00
test_pipelines.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00
test_scheduler.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00
test_training.py [ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334) 2022-09-05 14:49:26 +02:00