- 19 Aug, 2022 3 commits
-
-
Pedro Cuenca authored
* Implement `pipeline.to(device)` * DiffusionPipeline.to() decides best device on None. * Breaking change: torch_device removed from __call__ `pipeline.to()` now has PyTorch semantics. * Use kwargs and deprecation notice Co-authored-by:
Patrick von Platen <patrick.v.platen@gmail.com> * Apply torch_device compatibility to all pipelines. * style Co-authored-by:
Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by:
anton-l <anton@huggingface.co>
-
Suraj Patil authored
-
Suraj Patil authored
* add SafetyChecker * better name, fix checker * add checker in main init * remove from main init * update logic to detect pipeline module * style * handle all safety logic in safety checker * draw text * can't draw * small fixes * treat special care as nsfw * remove commented lines * update safety checker
-
- 17 Aug, 2022 2 commits
-
-
Anton Lozhkov authored
* Revive Make utils * Add datasets for training too
-
Suraj Patil authored
use default params in __call__
-
- 16 Aug, 2022 3 commits
-
-
Patrick von Platen authored
-
Anton Lozhkov authored
* test LMS with LDM * test LMS with LDM * Interchangeable sigma and timestep. Added dummy objects * Debug * cuda generator * Fix derivatives * Update tests * Rename Lms->LMS
-
Patrick von Platen authored
* [Half precision] Make sure half-precision is correct * Update src/diffusers/models/unet_2d.py * Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py * correct some tests * Apply suggestions from code review Co-authored-by:
Suraj Patil <surajp815@gmail.com> * finalize * finish Co-authored-by:
Suraj Patil <surajp815@gmail.com>
-
- 15 Aug, 2022 1 commit
-
-
Suraj Patil authored
* allow custom height width * raise if height width are not mul of 8
-
- 14 Aug, 2022 1 commit
-
-
Suraj Patil authored
* add stable diffusion pipeline * get rid of multiple if/else * batch_size is unused * add type hints * Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py * fix some bugs Co-authored-by:Patrick von Platen <patrick.v.platen@gmail.com>
-