Anton Lozhkov 8d9c4a531b [ONNX] Stable Diffusion exporter and pipeline (#399) ... * initial export and design * update imports * custom prover, import fixes * Update src/diffusers/onnx_utils.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/diffusers/onnx_utils.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * remove push_to_hub * Update src/diffusers/onnx_utils.py Co-authored-by: Suraj Patil <surajp815@gmail.com> * remove torch_device * numpify the rest of the pipeline * torchify the safety checker * revert tensor * Code review suggestions + quality * fix tests * fix provider, add an end-to-end test * style Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Suraj Patil <surajp815@gmail.com>		2022-09-08 15:17:28 +02:00
..
__init__.py	Fix conversion script	2022-07-15 17:00:41 +00:00
change_naming_configs_and_checkpoints.py	Style the `scripts` directory (#250)	2022-08-25 15:46:09 +02:00
conversion_ldm_uncond.py	Style the `scripts` directory (#250)	2022-08-25 15:46:09 +02:00
convert_ddpm_original_checkpoint_to_diffusers.py	Style the `scripts` directory (#250)	2022-08-25 15:46:09 +02:00
convert_ldm_original_checkpoint_to_diffusers.py	Style the `scripts` directory (#250)	2022-08-25 15:46:09 +02:00
convert_ncsnpp_original_checkpoint_to_diffusers.py	Style the `scripts` directory (#250)	2022-08-25 15:46:09 +02:00
convert_stable_diffusion_checkpoint_to_onnx.py	[ONNX] Stable Diffusion exporter and pipeline (#399)	2022-09-08 15:17:28 +02:00
generate_logits.py	[ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334)	2022-09-05 14:49:26 +02:00