6cf72a9b1e
Fix slow tests ( #1210 )
...
* fix tests
* Fix more
* more
2022-11-09 11:22:12 +01:00
24895a1f49
Fix cpu offloading ( #1177 )
...
* Fix cpu offloading
* get offloaded devices locally for SD pipelines
2022-11-09 10:28:10 +01:00
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
9d8943b7e7
Add CycleDiffusion pipeline using Stable Diffusion ( #888 )
...
* Add CycleDiffusion pipeline for Stable Diffusion
* Add the option of passing noise to DDIMScheduler
Add the option of providing the noise itself to DDIMScheduler, instead of the random seed generator.
* Update README.md
* Update README.md
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update scheduling_ddim.py
* Update import format
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update scheduling_ddim.py
* Update src/diffusers/schedulers/scheduling_ddim.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update src/diffusers/schedulers/scheduling_ddim.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update src/diffusers/schedulers/scheduling_ddim.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update src/diffusers/schedulers/scheduling_ddim.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update src/diffusers/schedulers/scheduling_ddim.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update scheduling_ddim.py
* Update scheduling_ddim.py
* Update scheduling_ddim.py
* add two tests
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Update README.md
* Rename pipeline name as suggested in the latest reviewer comment
* Update test_pipelines.py
* Update test_pipelines.py
* Update test_pipelines.py
* Update pipeline_stable_diffusion_cycle_diffusion.py
* Remove the generator
This generator does not control all randomness during sampling, which can be misleading.
* Update optimal hyperparameters
* Update src/diffusers/pipelines/stable_diffusion/README.md
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Update src/diffusers/pipelines/stable_diffusion/README.md
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Update src/diffusers/pipelines/stable_diffusion/README.md
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Apply suggestions from code review
* uP
* Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_cycle_diffusion.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* up
* up
* Replace assert with ValueError
* finish docs
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
2022-11-04 20:51:06 +01:00
1172c9634b
add enable sequential cpu offloading to other stable diffusion pipelines ( #1085 )
...
* add enable sequential cpu offloading to other stable diffusion pipelines
* trigger ci
* fix styling
* interpolate before converting to device to avoid breking when cpu_offload is enabled with fp16
Co-authored-by: Pedro Gengo <pedro.gabriel.lourenco@hotmail.com>
* style again I need to stop forgething this thing
* fix inpainting bug that could cause device misalignment
Co-authored-by: Pedro Gengo <pedro.gabriel.lourenco@hotmail.com>
* Apply suggestions from code review
Co-authored-by: Pedro Gengo <pedro.gabriel.lourenco@hotmail.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-11-04 19:25:28 +01:00
2c108693cc
Test precision increases ( #1113 )
...
* increase the precision of slice-based tests and make the default test case easier to single out
* increase precision of unit tests which already rely on float comparisons
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-11-04 17:54:01 +01:00
42bb459457
[Low cpu memory] Correct naming and improve default usage ( #1122 )
...
* correct naming
* finish
* Apply suggestions from code review
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
2022-11-03 18:11:18 +01:00
7482178162
default fast model loading 🔥 ( #1115 )
...
* make accelerate hard dep
* default fast init
* move params to cpu when device map is None
* handle device_map=None
* handle torch < 1.9
* remove device_map="auto"
* style
* add accelerate in torch extra
* remove accelerate from extras["test"]
* raise an error if torch is available but not accelerate
* update installation docs
* Apply suggestions from code review
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* improve defautl loading speed even further, allow disabling fats loading
* address review comments
* adapt the tests
* fix test_stable_diffusion_fast_load
* fix test_read_init
* temp fix for dummy checks
* Trigger Build
* Apply suggestions from code review
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
2022-11-03 17:25:57 +01:00
4a38166afe
Allow saving `None` pipeline components ( #1118 )
...
* Allow saving `None` pipeline components
* support flax as well
* style
2022-11-03 15:41:33 +01:00
4e59bcc680
[CI] Framework and hardware-specific CI tests ( #997 )
...
* [WIP][CI] Framework and hardware-specific docker images for CI tests
* username
* fix cpu
* try out the image
* push latest
* update workspace
* no root isolation for actions
* add a flax image
* flax and onnx matrix
* fix runners
* add reports
* onnxruntime image
* retry tpu
* fix
* fix
* build onnxruntime
* naming
* onnxruntime-gpu image
* onnxruntime-gpu image, slow tests
* latest jax version
* trigger flax
* run flax tests in one thread
* fast flax tests on cpu
* fast flax tests on cpu
* trigger slow tests
* rebuild torch cuda
* force cuda provider
* fix onnxruntime tests
* trigger slow
* don't specify gpu for tpu
* optimize
* memory limit
* fix flax tests
* disable docker cache
2022-11-02 14:07:07 +01:00
8ee21915bf
Integration tests precision improvement for inpainting ( #1052 )
...
* improve test precision
get tests passing with greater precision using lewington images
* make old numpy load function a wrapper around a more flexible numpy loading function
* adhere to black formatting
* add more black formatting
* adhere to isort
* loosen precision and replace path
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-11-02 11:47:26 +01:00
17c2c0600b
[Tests] Fix slow tests ( #1087 )
2022-10-31 18:59:58 +01:00
c18941b01a
[Better scheduler docs] Improve usage examples of schedulers ( #890 )
...
* [Better scheduler docs] Improve usage examples of schedulers
* finish
* fix warnings and add test
* finish
* more replacements
* adapt fast tests hf token
* correct more
* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* Integrate compatibility with euler
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-10-31 17:26:30 +01:00
a1ea8c01c3
k-diffusion-euler ( #1019 )
...
* k-diffusion-euler
* make style make quality
* make fix-copies
* fix tests for euler a
* Update src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
* Update src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
* Update src/diffusers/schedulers/scheduling_euler_discrete.py
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
* Update src/diffusers/schedulers/scheduling_euler_discrete.py
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
* remove unused arg and method
* update doc
* quality
* make flake happy
* use logger instead of warn
* raise error instead of deprication
* don't require scipy
* pass generator in step
* fix tests
* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* Update tests/test_scheduler.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* remove unused generator
* pass generator as extra_step_kwargs
* update tests
* pass generator as kwarg
* pass generator as kwarg
* quality
* fix test for lms
* fix tests
Co-authored-by: patil-suraj <surajp815@gmail.com>
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-10-31 16:20:38 +01:00
707b8684b3
fix slow test
2022-10-31 09:13:37 +00:00
8d6487f3cb
Fix some failing tests ( #1041 )
...
* up
* up
* up
* Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
* Apply suggestions from code review
2022-10-28 17:05:00 +02:00
d2d9764f35
[Tests] Speed up slow tests ( #1040 )
...
* [Tests] Speed up slow tests
* Up
* up
2022-10-28 14:46:39 +02:00
3be9fa97d6
[Accelerate model loading] Fix meta device and super low memory usage ( #1016 )
...
* [Accelerate model loading] Fix meta device and super low memory usage
* better naming
2022-10-27 12:11:42 +02:00
2c82e0c4eb
Reorganize pipeline tests ( #963 )
...
* Reorganize pipeline tests
* fix vq
2022-10-24 16:34:01 +02:00
25dfd0f8dc
[Tests] Move stable diffusion into their own files ( #936 )
...
* [Tests] Move stable diffusion into their own files
* up
2022-10-21 12:49:52 +02:00
Patrick von Platen