Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
diffusers/tests
History
Suraj Patil 9e234d8048
handle fp16 in `UNet2DModel` (#1216) 
* make sure fp16 runs well

* add fp16 test for superes

* Update src/diffusers/models/unet_2d.py

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

* gen on cuda

* always run fast inferecne test on cpu

* run on cpu

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
 		2022-11-23 11:13:34 +01:00
..
fixtures/custom_pipeline [Custom pipeline] Easier loading of local pipelines (#1327) 2022-11-17 16:00:26 +01:00
models Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
pipelines handle fp16 in `UNet2DModel` (#1216) 2022-11-23 11:13:34 +01:00
repo_utils Move accelerate to a soft-dependency (#1134) 2022-11-04 14:58:52 +01: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 Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_layers_utils.py VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
test_modeling_common.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_modeling_common_flax.py Flax: Trickle down `norm_num_groups` (#789) 2022-10-11 20:05:10 +02:00
test_outputs.py Fix BaseOutput initialization from dict (#570) 2022-09-20 18:32:16 +02:00
test_pipelines.py Fix using non-square images with UNet2DModel and DDIM/DDPM pipelines (#1289) 2022-11-23 11:11:39 +01:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py Flax tests: don't hardcode number of devices (#1175) 2022-11-09 20:04:43 +01:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_scheduler_flax.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01: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