Patrick von Platen 5c4ea00de7 Efficient Attention (#366) ...

* up

* add tests

* correct

* up

* finish

* better naming

* Update README.md

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

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

2022-09-06 18:06:47 +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

Efficient Attention (#366)

2022-09-06 18:06:47 +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