James R T 2558977bc7 Add callback parameters for Stable Diffusion pipelines (#521) ...

* Add callback parameters for Stable Diffusion pipelines

Signed-off-by: James R T <jamestiotio@gmail.com>

* Lint code with `black --preview`

Signed-off-by: James R T <jamestiotio@gmail.com>

* Refactor callback implementation for Stable Diffusion pipelines

* Fix missing imports

Signed-off-by: James R T <jamestiotio@gmail.com>

* Fix documentation format

Signed-off-by: James R T <jamestiotio@gmail.com>

* Add kwargs parameter to standardize with other pipelines

Signed-off-by: James R T <jamestiotio@gmail.com>

* Modify Stable Diffusion pipeline callback parameters

Signed-off-by: James R T <jamestiotio@gmail.com>

* Remove useless imports

Signed-off-by: James R T <jamestiotio@gmail.com>

* Change types for timestep and onnx latents

* Fix docstring style

* Return decode_latents and run_safety_checker back into __call__

* Remove unused imports

* Add intermediate state tests for Stable Diffusion pipelines

Signed-off-by: James R T <jamestiotio@gmail.com>

* Fix intermediate state tests for Stable Diffusion pipelines

Signed-off-by: James R T <jamestiotio@gmail.com>

Signed-off-by: James R T <jamestiotio@gmail.com>

2022-10-02 19:56:36 +02:00

..

__init__.py

fix issues with loading, add test for pipeline

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

conftest.py

[Tests] Upload custom test artifacts (#572)

2022-09-19 17:08:29 +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

[Tests] Fix spatial transformer tests on GPU (#531)

2022-09-16 16:04:37 +02:00

test_modeling_common.py

[UNet2DConditionModel] add gradient checkpointing (#461)

2022-09-22 15:36:47 +02:00

test_models_unet.py

[gradient checkpointing] lower tolerance for test (#652)

2022-09-29 11:57:37 +02:00

test_models_vae.py

Fix vae tests for cpu and gpu (#480)

2022-09-13 19:14:20 +02:00

test_models_vq.py

update expected results of slow tests (#268)

2022-09-12 15:49:39 +02:00

test_outputs.py

Fix BaseOutput initialization from dict (#570)

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

test_pipelines.py

Add callback parameters for Stable Diffusion pipelines (#521)

2022-10-02 19:56:36 +02:00

test_scheduler.py

Fix the LMS pytorch regression (#664)

2022-09-28 14:07:26 +02:00

test_training.py

[Pytorch] Pytorch only schedulers (#534)

2022-09-27 15:27:34 +02:00