Bryce Drennan
df86aa6668
feature: densepose controlnet ( #481 )
2024-04-04 22:02:25 -07:00
Bryce
76b6fa8b65
tests: fix them
2024-03-15 12:10:08 -07:00
Bryce
cf8a44b317
feature: update refiners
...
better handles img2img (partial diffusion runs)
2024-01-20 12:35:58 -08:00
Bryce
d2609cb5cd
fix: use smaller composition size
2024-01-06 17:23:27 -08:00
Bryce
32b5175e0e
feature: better upscaling
...
- use face enhancement in a smarter way that doesn't blur high-res images
- use a different upscale model for composition images
**Upscaling**
RealESRGAN is great but it blurs parts of images it doesn't understand
4xUltrasharp is a finetune of RealESRGan that isn't as good but doesn't have this blurry patch problem. This makes it more suitable to use as part of the composition/upscale process. We still use realesrgan for any last-step upscales since it does look better.
had to write a state dict translator to use the ultrasharp model
**Face Enhancement**
We no longer enhance faces that are larger than 512 pixels. They should already have enough details and the face enhancer doesn't produce faces at high enough resolution to look good at that size.
2023-12-21 05:48:02 -08:00
Bryce
0c03612d44
feature: large images now stay well-composed thanks to tile controlnet
2023-12-18 15:33:25 -08:00
jaydrennan
e7b6fc40fa
fix: adds default line ending for csv writing.
...
the csv library defaults to using CRLF line endings if not specified.
2023-12-14 21:12:25 -08:00
jaydrennan
3f3e080d39
feature: adds ability to use qrcode
...
feature: adds controlnet qrcode image generation.
feature: adds control net for qrcode image generation.
2023-12-14 21:12:25 -08:00
Bryce
62de446a92
ci: add mypy github action
2023-12-12 20:54:39 -08:00
Bryce
203747b14f
refactor: simplify model_weights/architecture
2023-12-12 20:54:39 -08:00
Bryce
2bd6cb264b
feature: large refactor
...
- add type hints
- size parameter
- ControlNetInput => ControlInput
- simplify imagineresult
2023-12-12 20:54:39 -08:00
Bryce
0fe3733933
fix: memory management issue
...
the dtype being used as a cache key wasn't consistent. this caused the model to be loaded twice
2023-12-03 09:13:01 -08:00
Bryce
b61d06651c
tests: fix tests
...
- disable details mode. needs more work done to support
2023-12-03 09:13:01 -08:00
Bryce
b5a0e65f35
fix: edit mode and some controlnet tests
2023-11-24 09:10:12 -08:00
Bryce
f97f6a3b4b
feature: use refiners library for generation
...
BREAKING CHANGE
- stable diffusion 1.5 + inpainting working
- self-attention guidance working. improves image generation quality
- tile-mode working
- inpainting self-attention guidance working
disable/broken features:
- sd 1.4, 2.0, 2.1
- most of the samplers
- pix2pix edit
- most of the controlnets
- memory management
- python 3.8 support
wip
2023-11-22 13:22:00 -08:00
Bryce
82d74c6b49
feature: switch to pydantic models
...
- allow prompt re-use by deferring random seed
2023-05-31 20:02:09 -07:00
Bryce
38ac0b7f54
tests: fix tests
2023-05-20 13:09:00 -07:00
Bryce
df25936d6f
feature: automatic use of inpainting
...
feature disabled since controlnet inpainting doesn't work great. Was disabled by setting `inpaint_method="finetune",`
2023-05-20 13:09:00 -07:00
Bryce
6db296aa37
tests: faster tests
2023-05-20 11:35:49 -07:00
Bryce
c082ea523f
tests: update tests
...
- controlnet version changes + graphics card change
2023-05-13 23:41:00 -07:00
Bryce
db8d3b74ea
feature: add "detail" control mode.
...
Add details to an image. Great for upscaling an image.
2023-05-05 02:53:43 -07:00
Bryce
476a81a967
feature: add "shuffle" control mode
...
Image is generated from elements of control image. similar to style transfer
2023-05-05 02:53:43 -07:00
Bryce
832adf27bc
feature: update controlnets to 1.1
...
- smaller filesizes for the controlnet weights
- fix: works with other 1.5 based models
- enables more control modes getting added
2023-05-04 19:48:27 -07:00
Bryce
3b777b98d8
fix: improved large images using composition
2023-02-28 21:04:00 -08:00
Bryce
40da2b0357
tests: fix tests
2023-02-28 21:04:00 -08:00
Bryce
70c58467c0
fix: img2img was broken for all samplers except ddim,plms
...
img2img was broken for all samplers except plms and ddim when init image strength was >~0.25. Been this way for a while. whoops
2023-02-28 21:04:00 -08:00
Bryce
54c3ad51d6
feature: controlnet
2023-02-22 23:38:47 -08:00
Bryce
8a97213622
feature: sliced latent decoding
...
allows generation of bigger images. tile seams can be noticeable occasionally despite the feathering
2023-02-12 01:41:57 -08:00
Bryce Drennan
542e4fbd55
fix: tile mode was broken since latest perf improvements ( #220 )
2023-01-27 22:56:46 -08:00
Bryce Drennan
7deabaae62
tests: update expected images for tests ( #215 )
2023-01-26 20:41:47 -08:00
Bryce
8855823293
fix: rename files for windows compat
2023-01-22 19:08:01 -08:00
Bryce
81f294216b
feature: 🎉 outpainting
2023-01-17 00:31:01 -08:00
Bryce
9e0a9e2c29
tests: update
2023-01-02 14:11:36 -08:00
Bryce
da0f1e1ee6
fix: make sure randomness is generated on cpu for consistency
2022-12-21 09:33:46 -08:00
Bryce
1f7403155e
fix: more thorough cleaning of memory when switching models
...
also cleanup up some test failures
2022-12-18 22:00:29 -08:00
Bryce
f30823e0b5
feature: Stable Diffusion 2.1
2022-12-07 10:41:36 -08:00
Bryce
2f8edc1530
feature: negative prompting
...
- feature: negative prompting. `--negative-prompt` or `ImaginePrompt(..., negative_prompt="ugly, deformed, extra arms, etc")`
- feature: a default negative prompt is added to all generations. Images in SD-2.0 don't look bad anymore. Images in 1.5 look improved as well.
2022-12-02 02:03:13 -08:00
Bryce
554fc1d0ec
tests: update images
...
- probably the k-diffusin update fixed some bugs
- add missing images
2022-11-28 00:32:53 -08:00
Bryce
9c153a8d2d
feature: support k-sampler methods for SD-2.0-v (768)
2022-11-26 16:18:08 -08:00
Bryce
015088507f
feature/refactor/fix: better defaults. correct version in metadata
...
- feature: use different default steps and image sizes depending on sampler and model selceted
- fix : #110 use proper version in image metadata
- refactor: samplers all have their own class that inherits from ImageSampler
2022-11-26 16:18:08 -08:00
Bryce
7fba2972e8
feature: added DPM++ 2S a
and DPM++ 2M
samplers
...
-fix: fix bug with `--show-work`
2022-11-11 10:52:48 -08:00
Bryce
021a0c540d
feature: inpainting model support; improved model manager
2022-10-25 20:07:35 -07:00
Bryce
94c0aeab12
feature: switch between models without restarting python instance
2022-10-23 22:55:28 -07:00
Bryce
4d802a043e
feature: stable diffusion 1.5
2022-10-22 02:22:57 -07:00
Bryce
8332593fed
feature: dilation and erosion of masks
...
Previously the `+` and `-` characters in a mask (example: `face{+0.1}`) added to the grayscale value of any masked areas. This wasn't very useful. The new behavior is that the mask will expand or contract by the number of pixel specified. The technical terms for this are dilation and erosion. This allows much greater control over the masked area.
2022-10-17 23:47:26 -07:00
Bryce
741a433c56
feature: update k_diffusion. add dpm_fast and dpm_addaptive samplers
2022-10-16 22:10:54 -07:00
Bryce
dcf953383e
tests: support distributed test runs
...
- switch to expected images instead of hashes. allow fuzzy matches
feature: more consistent seeds
2022-10-16 20:22:16 -07:00