Commit Graph

350 Commits (master)

Author SHA1 Message Date
Bryce 1faea372f9 fix: cleanup logging - remove unnecessary version checks 4 weeks ago
jaydrennan cc79cac5fc refactor: ruff formatting 4 weeks ago
jaydrennan 964dd4ead7 feature: integrates spandrel for upscaling 4 weeks ago
Bryce efbab3a141 fix: be forgiving of - vs _ in video decoder model name
Addresses https://github.com/brycedrennan/imaginAIry/issues/485
4 weeks ago
Bryce 3a9a3974ce fix: allow referencing local paths for sdxl model weights
Addresses https://github.com/brycedrennan/imaginAIry/issues/484
4 weeks ago
Bryce Drennan 3c1c695f76
feature: cloth segmentation (#482) 1 month ago
Bryce Drennan df86aa6668
feature: densepose controlnet (#481) 1 month ago
Bryce Drennan 49f2c25b6b
feature: IP-Adapter (#477)
todo
- allow specification ip adapter weights/arch


---------

Co-authored-by: jaydrennan <jsdman1313@gmail.com>
2 months ago
Bryce 9c48b749d8 feature: script for running imaginairy in the modal.com cloud 2 months ago
Bryce 9cdacd454f style: use latest ruff 2 months ago
Bryce e6a1c988c5 fix: if weights are float32 but float16 was specified, still use float16 4 months ago
Bryce cf8a44b317 feature: update refiners
better handles img2img (partial diffusion runs)
4 months ago
jaydrennan 1bf53e47cf
feature: updates refiners vendored library (#458)
* feature: updates refiners vendored library

has a small bugfix that will soon be replaced by a better fix from upstream refiners

Co-authored-by: Bryce <github20210803@accounts.brycedrennan.com>
4 months ago
Bryce Drennan 601a112dc3
refactor: move download related functions to separate module (#453)
+ renames and typehints
4 months ago
Bryce Drennan 502ffbdc63
feature: sdxl inpaint support (#450) 4 months ago
Bryce Drennan 700cb457b9
feature: support loading sdxl compvis weights (#449) 4 months ago
Bryce Drennan 907e80d1f2
feature: video interpolation (#448)
- uses rife algorithm to interpolate frames
4 months ago
Bryce Drennan bb2dd45cf2
feature: videogen improvements (#447)
- allow generation at any size
- output "bounce" animations
- choose output format: mp4, webp, or gif
- fix random seed handling
- organize some code better
4 months ago
Bryce d2609cb5cd fix: use smaller composition size 4 months ago
Bryce 5bbb09f69e build: vendorize facexlib
had too many unused sub-dependencies

also monkeypatch the download mechanism to use our standard download function
4 months ago
Bryce Drennan d3106fc9e3
fix: videogen bug (#443) 4 months ago
jaydrennan 89bc1a9f1c
docs: adds docs tool, material for mkdocs, along with more fleshed ou… (#428)
* docs: adds docs tool, material for mkdocs, along with more fleshed out docstrings.

this includes ability to serve up a local docs website.


---------

Co-authored-by: Bryce <github20210803@accounts.brycedrennan.com>
4 months ago
Bryce Drennan 0271bffa38
build: remove fairscale dependency (#441) 4 months ago
Bryce e00c7b9eb7 fix: add back missing init file 4 months ago
Bryce d148bc1537 fix: progress latent collection bug 4 months ago
Bryce ed40a12c01 version: 14.0.0 4 months ago
Bryce 57dc27df8c build: tag imaginairy as typed 4 months ago
Bryce 55e27160f5 build: vendorize refiners
so we can still work in conda envs
4 months ago
Bryce f84406f12c fix: handle unexpected keys in weights better 4 months ago
Bryce 5b3b04b877 build: remove pytorch lightning dependency 4 months ago
Bryce 7100d3f9ea perf: make upscaler use fp16 for better efficiency 5 months ago
Bryce 4fcfc363af fix: always show total last 5 months ago
Bryce f50a1f5b0c fix: interrupted generations don't prevent more generations
fixes #424

- pref: improve memory usage when loading SD15.
- feature: clean up CLI output more
- feature: cuda memory tracking context manager
- feature: use safetensors fp16 for sd15
5 months ago
Bryce 9e3403df89 feature: clean up terminal output
- recording timing and memory usage of various steps
- re-use logging context for composition images
- load sdxl weights in a more VRAM efficient way
- switch to diffusers weights for default weights for sd15
5 months ago
Bryce Drennan 77c4b85037
perf: improve memory usage (#433)
add warning for corrupt weights files
5 months ago
Bryce Drennan 42a045e8e6
feature: support sdxl (#431)
- adds support for (SDXL)[https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0]
  - adds sliced encoding/decoding to refiners sdxl pipeline
  - doesn't support inpainting, controlnets
- monkeypatches self_attention_guidance to use sliced attention
- adds a bunch of model weight translation utilities and weightmaps
- add [opendalle 1.1](https://huggingface.co/dataautogpt3/OpenDalleV1.1)
- change default model to opendalle
- fix: better handle special characters in path inputs on command line
**todo**
- add tests
5 months ago
jaydrennan 3322777f5e refactor: formatting changes ruff formatter
ruff formatter has slight differences in formatting compared to black
5 months ago
jaydrennan 7eef3bf628 feature: replaces black formatter with ruff formatter 5 months ago
Bryce Drennan a2c38b3ec0
feature: support loading diffusers folder/based models from huggingface (#427) 5 months ago
Bryce 50e796a3b7 refactor: move code around 5 months ago
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.
5 months ago
Bryce 6ebd12abb1 refactor: move code to more intuitive places 5 months ago
Bryce 8cfb46d6de fix: bug in sliced encoder 5 months ago
Bryce 372453e645 refactor: remove training code 5 months ago
Bryce Drennan 616f686ed2
small changes (#425)
* docs: update todo

* refactor: small cleanup of tiling code
5 months ago
jaydrennan df00109074 refactor: space formatting 5 months ago
jaydrennan 0c01cd690f
fix: sets correct default value for composition strength. (#422)
also corrects positional argument error by requiring _imagine_cmd to take key word arguments.
5 months ago
Bryce 0c03612d44 feature: large images now stay well-composed thanks to tile controlnet 5 months ago
jaydrennan 2372a71e6c fix: adds tile/detail controlnet back in. 5 months ago
Bryce Drennan f88b5c1b2b
fix: word images still work without specified size (#421) 5 months ago