Sayak Paul 7b100ce589 [Tests] conditionally check fp8_e4m3_bf16_max_memory < fp8_e4m3_fp32_max_memory (#10669) ...

* conditionally check if compute capability is met.

* log info.

* fix condition.

* updates

* updates

* updates

* updates

2025-01-28 12:00:14 +05:30

..

autoencoders

[core] Layerwise Upcasting (#10347)

2025-01-22 19:49:37 +05:30

transformers

fix offload gpu tests etc (#10366)

2025-01-21 18:52:36 +05:30

unets

[core] Layerwise Upcasting (#10347)

2025-01-22 19:49:37 +05:30

__init__.py

[Tests] Improve unet / vae tests (#1018)

2022-10-28 13:43:26 +02:00

test_activations.py

move activation dispatches into helper function (#3656)

2023-06-05 12:30:48 -07:00

test_attention_processor.py

[Tests] Fix more tests sayak (#10359)

2024-12-23 19:11:21 +05:30

test_layers_utils.py

Support bfloat16 for Upsample2D (#9480)

2024-10-01 16:08:12 -10:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

[Tests] conditionally check fp8_e4m3_bf16_max_memory < fp8_e4m3_fp32_max_memory (#10669)

2025-01-28 12:00:14 +05:30