a4d5b59f13
Refactor Pipelines / Community pipelines and add better explanations. ( #257 )
...
* [Examples readme]
* Improve
* more
* save
* save
* save more
* up
* up
* Apply suggestions from code review
Co-authored-by: Nathan Lambert <nathan@huggingface.co>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* up
* make deterministic
* up
* better
* up
* add generator to img2img pipe
* save
* make pipelines deterministic
* Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
* apply all changes
* more correctnios
* finish
* improve table
* more fixes
* up
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
* Update src/diffusers/pipelines/README.md
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* add better links
* fix more
* finish
Co-authored-by: Nathan Lambert <nathan@huggingface.co>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
2022-08-30 18:43:42 +02:00
5e84353eba
Refactor progress bar ( #242 )
...
* Refactor progress bar of pipeline __call__
* Make any tqdm configs available
* remove init
* add some tests
* remove file
* finish
* make style
* improve progress bar test
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-08-30 12:30:06 +02:00
efa773afd2
Support K-LMS in img2img ( #270 )
...
* Support K-LMS in img2img
* Apply review suggestions
2022-08-29 17:17:05 +02:00
da7d4cf200
[BugFix]: Fixed add_noise in LMSDiscreteScheduler ( #253 )
...
* Fixed add_noise in LMSDiscreteScheduler
* Linting
* Update src/diffusers/schedulers/scheduling_lms_discrete.py
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
2022-08-29 16:40:49 +02:00
9e1b1ca49d
[Tests] Make sure tests are on GPU ( #269 )
...
* [Tests] Make sure tests are on GPU
* move more models
* speed up tests
2022-08-29 15:58:11 +02:00
16172c1c7e
Adds missing torch imports to inpainting and image_to_image example ( #265 )
...
adds missing torch import to example
2022-08-29 10:56:37 +02:00
28f730520e
Fix typo in README.md ( #260 )
2022-08-26 18:54:45 -07:00
5cbed8e0d1
Fix inpainting script ( #258 )
...
* expand latents before the check, style
* update readme
2022-08-26 21:16:43 +05:30
11133dcca1
Initialize CI for code quality and testing ( #256 )
...
* Init CI
* clarify cpu
* style
* Check scripts quality too
* Drop smi for cpu tests
* Run PR tests on cpu docker envs
* Update .github/workflows/push_tests.yml
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Try minimal python container
* Print env, install stable GPU torch
* Manual torch install
* remove deprecated platform.dist()
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-08-26 17:34:58 +02:00
bb4d605dfc
add inpainting example script ( #241 )
...
* add inpainting
* added proper noising of init_latent as reccommened by jackloomen (https://github.com/huggingface/diffusers/pull/241#issuecomment-1226283542 )
* move image preprocessing inside pipeline and allow non 512x512 mask
2022-08-26 20:32:46 +05:30
e5b5deaea6
Update README.md with examples ( #252 )
...
* Update README.md
* Update README.md
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* Update README.md
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-08-26 15:27:25 +05:30
bfe37f3159
Reproducible images by supplying latents to pipeline ( #247 )
...
* Accept latents as input for StableDiffusionPipeline.
* Notebook to demonstrate reusable seeds (latents).
* More accurate type annotation
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Review comments: move to device, raise instead of assert.
* Actually commit the test notebook.
I had mistakenly pushed an empty file instead.
* Adapt notebook to Colab.
* Update examples readme.
* Move notebook to personal repo.
Co-authored-by: Suraj Patil <surajp815@gmail.com>
2022-08-25 19:17:05 +05:30
89793a97e2
Style the `scripts` directory ( #250 )
...
Style scripts
2022-08-25 15:46:09 +02:00
365f75233f
Pin black==22.3 to keep a stable --preview flag ( #249 )
...
Pin black==22.3
2022-08-25 15:19:59 +02:00
c1efda70b5
[Clean up] Clean unused code ( #245 )
...
* CleanResNet
* refactor more
* correct
2022-08-25 15:25:57 +05:30
47893164ab
added test workflow and fixed failing test ( #237 )
...
* added test workflow and fixed failing test
* 4 decimal places
2022-08-24 13:46:53 +02:00
102cabeb23
split tests_modeling_utils ( #223 )
...
* split tests_modeling_utils
* Fix SD tests .to(device)
* fix merge
* Fix style
Co-authored-by: anton-l <anton@huggingface.co>
2022-08-24 13:27:16 +02:00
511bd3aaf2
[example/image2image] raise error if strength is not in desired range ( #238 )
...
raise error if strength is not in desired range
2022-08-23 19:52:52 +05:30
4674fdf807
Add image2image example script. ( #231 )
...
* boom boom
* reorganise examples
* add image2image in example inference
* add readme
* fix example
* update colab url
* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* fix init_timestep
* update colab url
* update main readme
* rename readme
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-08-23 16:27:28 +05:30
6028d58cb0
Remove dead code in `resnet.py` ( #218 )
...
remove dead code in resnet.py
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-08-23 12:08:37 +05:30
60a147343f
Release: v0.2.4
2022-08-22 18:45:43 +02:00
eb5267f377
Style quickfix
2022-08-22 18:40:04 +02:00
db5fa43079
[Loading] allow modules to be loaded in fp16 ( #230 )
2022-08-22 18:27:17 +02:00
0ab948568d
Add more visibility to the colab links with badges
2022-08-22 14:15:24 +02:00
ebd80e2618
Release: v0.2.3
2022-08-22 10:49:38 +02:00
89509230db
Merge remote-tracking branch 'origin/main'
2022-08-22 10:22:36 +02:00
577a6a65d6
Fix SD tests .to(device)
2022-08-22 10:22:28 +02:00
62b3efe351
Fix SD example typo
2022-08-22 09:25:55 +02:00
21ceda3f6c
Remove duplicate add_noise
2022-08-22 09:12:42 +02:00
5321f3e203
add add_noise method in LMSDiscreteScheduler, PNDMScheduler ( #227 )
...
add add_noise method in more schedulers
2022-08-22 08:38:07 +02:00
3f1861ee46
hotfix for pdnm test ( #220 )
2022-08-22 07:23:59 +02:00
6a03060c45
Restore `is_modelcards_available` in `.utils` ( #224 )
...
Restore `is_modelcards_available` in `.utils`.
Otherwise attempting to import `hub_utils` (in training scripts, for
example), fails.
This was removed during the refactor in df90f0c
2022-08-22 07:21:29 +02:00
2b7669183e
Update README for 0.2.3 release ( #225 )
...
* Update README for 0.2.3 release:
* Apply suggestions from code review
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-08-21 23:59:46 +02:00
e7b69cbe19
[Safety Checker] Lower adjustment value
2022-08-21 15:29:10 +00:00
3cde81408f
Add incompatibility note for SD (temporary)
2022-08-20 12:02:32 +02:00
71ba8aec55
Pipeline to device ( #210 )
...
* 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>
2022-08-19 18:39:08 +02:00
89e9521048
fix safety check ( #217 )
2022-08-19 18:04:58 +05:30
65ea7d6b62
Add safety module ( #213 )
...
* 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
2022-08-19 15:24:03 +05:30
e30e1b89d0
Support one-string prompts and custom image size in LDM ( #212 )
...
* Support one-string prompts in LDM
* Add other features from SD too
2022-08-18 17:55:15 +02:00
df90f0ce98
Add `is_torch_available`, `is_flax_available` ( #204 )
...
* Add is_<framework>_available, refactor import utils
* deps
* quality
2022-08-17 16:47:20 +02:00
ed22b4fd07
Revive `make quality` ( #203 )
...
* Revive Make utils
* Add datasets for training too
2022-08-17 15:22:04 +02:00
f9522d825c
[StableDiffusionPipeline] use default params in __call__ ( #196 )
...
use default params in __call__
2022-08-17 17:06:12 +05:30
80e0c8ba9e
fix stable-diffusion code snippet format.
2022-08-17 14:15:00 +05:30
3cd20d59d7
fix test_from_pretrained_hub_pass_model ( #194 )
...
init pipeline once
2022-08-17 13:58:18 +05:30
e36a36788e
Match params with official Stable Diffusion lib ( #192 )
...
https://github.com/CompVis/stable-diffusion
2022-08-16 22:52:22 +02:00
4b02f53e62
Release: v0.2.2
2022-08-16 19:30:08 +02:00
27d11a0094
[K-LMS Scheduler] fix import ( #191 )
2022-08-16 19:25:45 +02:00
554e67cb06
Update README.md
2022-08-16 19:12:25 +02:00
45cb500667
Update README.md
2022-08-16 19:10:35 +02:00
8c78e73fef
Update README.md
2022-08-16 19:09:09 +02:00
Patrick von Platen