Pedro Cuenca dc277501c7 Flax memory efficient attention (#2889) ...

* add use_memory_efficient params placeholder

* test

* add memory efficient attention jax

* add memory efficient attention jax

* newline

* forgot dot

* Rename use_memory_efficient

* Keep dtype last.

* Actually use key_chunk_size

* Rename symbol

* Apply style

* Rename use_memory_efficient

* Keep dtype last

* Pass `use_memory_efficient_attention` in `from_pretrained`

* Move JAX memory efficient attention to attention_flax.

* Simple test.

* style

---------

Co-authored-by: muhammad_hanif <muhammad_hanif@sofcograha.co.id>
Co-authored-by: MuhHanif <48muhhanif@gmail.com>

2023-04-12 10:17:51 +01:00

..

fixtures

Fix config prints and save, load of pipelines (#2849)

2023-04-11 13:35:42 +02:00

models

add only cross attention to simple attention blocks (#3011)

2023-04-11 14:38:50 -07:00

pipelines

Attn added kv processor torch 2.0 block (#3023)

2023-04-11 16:54:22 -07:00

repo_utils

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

schedulers

tests and additional scheduler fixes

2023-04-10 12:59:33 -07:00

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

conftest.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

pipeline_params.py

Add AudioLDM (#2232)

2023-03-23 19:00:21 +01:00

test_config.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_ema.py

[Tests] Correct PT2 (#2724)

2023-03-18 18:38:04 +01:00

test_hub_utils.py

Remove modelcards dependency (#2050)

2023-01-20 16:39:42 +01:00

test_image_processor.py

Add image_processor (#2617)

2023-03-15 07:55:49 -10:00

test_layers_utils.py

fix slow tsets (#3066)

2023-04-12 10:23:52 +02:00

test_lora_layers.py

[LoRA] Enabling limited LoRA support for text encoder (#2918)

2023-04-12 08:29:04 +05:30

test_modeling_common_flax.py

Allow to set config params directly in init (#1419)

2022-11-25 15:07:09 +01:00

test_modeling_common.py

[Init] Make sure shape mismatches are caught early (#2847)

2023-03-28 09:08:28 +01:00

test_outputs.py

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_pipelines_common.py

mps: remove warmup passes (#2771)

2023-03-22 19:29:27 +01:00

test_pipelines_flax.py

Flax memory efficient attention (#2889)

2023-04-12 10:17:51 +01:00

test_pipelines_onnx_common.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_pipelines.py

Fix config prints and save, load of pipelines (#2849)

2023-04-11 13:35:42 +02:00

test_training.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_unet_2d_blocks.py

Fix typo and format BasicTransformerBlock attributes (#2953)

2023-04-12 00:31:05 +02:00

test_unet_blocks_common.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_utils.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00