Anton Lozhkov
|
11f7d6f3cc
|
[ONNX] Improve ONNXPipeline scheduler compatibility, fix safety_checker (#1173)
* [ONNX] Improve ONNX scheduler compatibility, fix safety_checker
* typo
|
2022-11-08 14:39:11 +01:00 |
|
SkyTNT
|
0b42b074b4
|
[Onnx] support half-precision and fix bugs for onnx pipelines (#932)
* [Onnx] support half-precision and fix bugs for onnx pipelines
* Update convert_stable_diffusion_checkpoint_to_onnx.py
* style
* fix has_nsfw_concept
* Update convert_stable_diffusion_checkpoint_to_onnx.py
* fix style
|
2022-10-25 16:48:53 +02:00 |
|
Anton Lozhkov
|
89d124945a
|
ONNX supervised inpainting (#906)
* ONNX supervised inpainting
* sync with the torch pipeline
* fix concat
* update ref values
* back to 8 steps
* type fix
* make fix-copies
|
2022-10-19 17:03:31 +02:00 |
|
Anton Lozhkov
|
8eb9d9703d
|
Improve ONNX img2img numpy handling, temporarily fix the tests (#899)
* [WIP] Onnx img2img determinism
* more numpy + seed
* numpy inpainting, tolerance
* revert test workflow
|
2022-10-19 11:26:32 +02:00 |
|
Žilvinas Ledas
|
a9908ecfc1
|
Stable Diffusion image-to-image and inpaint using onnx. (#552)
* * Stabe Diffusion img2img using onnx.
* * Stabe Diffusion inpaint using onnx.
* Export vae_encoder, upgrade img2img, add test
* updated inpainting pipeline + test
* style
Co-authored-by: anton-l <anton@huggingface.co>
|
2022-10-18 17:44:01 +02:00 |
|
Justin Chu
|
75bb6d2d46
|
Fix ONNX conversion script opset argument type (#739)
The opset argument should be an `int` but was set as a `str`.
|
2022-10-07 15:47:43 +02:00 |
|
Patrick von Platen
|
78744b6a8f
|
No more use_auth_token=True (#733)
* up
* uP
* uP
* make style
* Apply suggestions from code review
* up
* finish
|
2022-10-05 17:16:15 +02:00 |
|
Anton Lozhkov
|
6bd005ebbe
|
[ONNX] Collate the external weights, speed up loading from the hub (#610)
|
2022-09-21 22:26:30 +02:00 |
|
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 |