Patrick von Platen 7c2262640b Align PT and Flax API - allow loading checkpoint from PyTorch configs (#827) ...

* up

* finish

* add more tests

* up

* up

* finish

2022-10-13 17:43:06 +02:00

..

fixtures/custom_pipeline

Custome Pipelines (#744)

2022-10-06 16:54:02 +02:00

__init__.py

fix issues with loading, add test for pipeline

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

conftest.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +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

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_modeling_common_flax.py

Flax: Trickle down norm_num_groups (#789)

2022-10-11 20:05:10 +02:00

test_modeling_common.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_models_unet.py

Fix gradient checkpointing test (#797)

2022-10-10 19:40:33 +02:00

test_models_vae_flax.py

Flax: Trickle down norm_num_groups (#789)

2022-10-11 20:05:10 +02:00

test_models_vae.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_models_vq.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_outputs.py

Fix BaseOutput initialization from dict (#570)

2022-09-20 18:32:16 +02:00

test_pipelines_flax.py

Align PT and Flax API - allow loading checkpoint from PyTorch configs (#827)

2022-10-13 17:43:06 +02:00

test_pipelines.py

Give more customizable options for safety checker (#815)

2022-10-13 15:52:26 +02:00

test_scheduler.py

[Scheduler design] The pragmatic approach (#719)

2022-10-05 14:41:19 +02:00

test_training.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_utils.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00