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 |