Commit Graph

529 Commits (df86aa66680dda6c84a39d51fc3882d7ffcfaf42)
 

Author SHA1 Message Date
Bryce Drennan 77c4b85037
perf: improve memory usage (#433)
add warning for corrupt weights files
6 months ago
Bryce 26d1ff9bc4 version: 14.0.0b6 6 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
6 months ago
jaydrennan fb7cff3684
Merge pull request #430 from brycedrennan/ruff_formatter
Ruff formatter
6 months ago
jaydrennan 3322777f5e refactor: formatting changes ruff formatter
ruff formatter has slight differences in formatting compared to black
6 months ago
jaydrennan 7eef3bf628 feature: replaces black formatter with ruff formatter 6 months ago
Bryce Drennan a2c38b3ec0
feature: support loading diffusers folder/based models from huggingface (#427) 6 months ago
Bryce 50e796a3b7 refactor: move code around 6 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.
6 months ago
Bryce 6ebd12abb1 refactor: move code to more intuitive places 6 months ago
Bryce 8cfb46d6de fix: bug in sliced encoder 6 months ago
Bryce 372453e645 refactor: remove training code 6 months ago
Bryce Drennan 616f686ed2
small changes (#425)
* docs: update todo

* refactor: small cleanup of tiling code
6 months ago
Bryce d834e8b5b3 test: run non-gpu tests on github 6 months ago
jaydrennan df00109074 refactor: space formatting 6 months ago
jaydrennan d39486af54 fix: updates test marking to use nodeid instead of name 6 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.
6 months ago
Bryce 0c03612d44 feature: large images now stay well-composed thanks to tile controlnet 6 months ago
jaydrennan 2372a71e6c fix: adds tile/detail controlnet back in. 6 months ago
Bryce Drennan f88b5c1b2b
fix: word images still work without specified size (#421) 7 months ago
Bryce 7880ee1389 feature: update midas (depth maps) 7 months ago
Bryce bf14ee6ee6 feature: add christmas-scene phrase list
Also add script that uses chatgpt to generate phrase-lists
7 months ago
Bryce c6ac5f553a refactor: separate controlnet image preprocessing 7 months ago
Bryce 9a0e0cd1a7 feature: better depth maps 7 months ago
jaydrennan 2eee741b20
Merge pull request #418 from brycedrennan/fix_test_set_spu_full
fix: configures test_set_gpu_full to run on a m1 mac.
7 months ago
jaydrennan c99a169986 fix: configures test_set_gpu_full to run on a m1 mac. 7 months ago
Bryce Drennan 2144f26fa7
feature: add ability to dynamically make word images (#417) 7 months ago
jaydrennan 3bd3dfdeaf
feature: adds --composition-strength parameter to cli (#416) 7 months ago
jaydrennan 41a9d7007b
test: adds tests for stablestudio (#415) 7 months ago
jaydrennan e1e6f8037c
refactor: removes unused code and configurations (#405)
Co-authored-by: jaydrennan
7 months ago
Bryce 6d39d791b1 refactor: move safety to utils 7 months ago
Bryce 168a843f29 refactor: move colorize to api 7 months ago
Bryce 96f4268d44 refactor: move video_sample to api 7 months ago
Bryce e72e8992ab refactor: create api module 7 months ago
Bryce ad561e8833 refactor: move model_manager to utils 7 months ago
Bryce d478771cc0 refactor: move a bunch of stuff to utils 7 months ago
Bryce 987af23abe refactor: move train.py 7 months ago
Bryce 0c456cd52a refactor: remove lr_scheduler.py 7 months ago
Bryce 01e32ff3f6 refactor: move bin files 7 months ago
Bryce 316114e660 docs: add docstrings
Wrote an openai script and custom prompt to generate them.
7 months ago
jaydrennan e7b6fc40fa fix: adds default line ending for csv writing.
the csv library defaults to using CRLF line endings if not specified.
7 months ago
jaydrennan 3f3e080d39 feature: adds ability to use qrcode
feature: adds controlnet qrcode image generation.
feature: adds control net for qrcode image generation.
7 months ago
Bryce 62de446a92 ci: add mypy github action 7 months ago
Bryce 012cc648d3 style: fix all the mypy typing issues
...or ignore them.
7 months ago
Bryce 5a636e45c5 feature: skip composition at sizes slightly larger than model is expecting 7 months ago
Bryce 203747b14f refactor: simplify model_weights/architecture 7 months ago
Bryce 37ecd1e5e0 fix: videogen. track gpu tests 7 months ago
Bryce eae4f20ae2 ci: add type checker
fix some typehint issues
7 months ago
Bryce e898e3a799 fix: several cli commands, edit demo, negative prompt
- fix colorize cmd. add test
- fix describe cmd. add test
- fix model-list cmd. add test
- fix stable studio
- hide stack grace for ValueErrors in cli
- set controlnet scale
- fix negative prompt to allow emptystring instead of replacing it with default
- adjust edit-demo parameters
- arg scheduler that works at click level (but disable it). works but not ideal experience.
7 months ago
Bryce c299cfffd9 fix: cache the controlnet models 7 months ago