Patrick von Platen
a7ca03aa85
Replace flake8 with ruff and update black ( #2279 )
...
* before running make style
* remove left overs from flake8
* finish
* make fix-copies
* final fix
* more fixes
2023-02-07 23:46:23 +01:00
Patrick von Platen
29b2c93c90
Make repo structure consistent ( #1862 )
...
* move files a bit
* more refactors
* fix more
* more fixes
* fix more onnx
* make style
* upload
* fix
* up
* fix more
* up again
* up
* small fix
* Update src/diffusers/__init__.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* correct
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-12-30 11:51:08 +01:00
Anton Lozhkov
86aa747da9
Fix ONNX conversion and inference ( #1416 )
2022-11-25 14:51:17 +01:00
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