Commit Graph

537 Commits (master)
 

Author SHA1 Message Date
Bryce 9f33fa0664 version: 14.0.0b9 5 months ago
Bryce 35a5ccbcba fix: include weightmaps 5 months ago
Bryce e90bf1b47d version: 14.0.0b8 5 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 0d78b8271f version: 14.0.0b7 5 months ago
Bryce Drennan 77c4b85037
perf: improve memory usage (#433)
add warning for corrupt weights files
5 months ago
Bryce 26d1ff9bc4 version: 14.0.0b6 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 fb7cff3684
Merge pull request #430 from brycedrennan/ruff_formatter
Ruff formatter
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
Bryce d834e8b5b3 test: run non-gpu tests on github 5 months ago
jaydrennan df00109074 refactor: space formatting 5 months ago
jaydrennan d39486af54 fix: updates test marking to use nodeid instead of name 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
Bryce 7880ee1389 feature: update midas (depth maps) 5 months ago
Bryce bf14ee6ee6 feature: add christmas-scene phrase list
Also add script that uses chatgpt to generate phrase-lists
5 months ago
Bryce c6ac5f553a refactor: separate controlnet image preprocessing 5 months ago
Bryce 9a0e0cd1a7 feature: better depth maps 5 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.
5 months ago
jaydrennan c99a169986 fix: configures test_set_gpu_full to run on a m1 mac. 5 months ago
Bryce Drennan 2144f26fa7
feature: add ability to dynamically make word images (#417) 5 months ago
jaydrennan 3bd3dfdeaf
feature: adds --composition-strength parameter to cli (#416) 5 months ago
jaydrennan 41a9d7007b
test: adds tests for stablestudio (#415) 5 months ago
jaydrennan e1e6f8037c
refactor: removes unused code and configurations (#405)
Co-authored-by: jaydrennan
5 months ago
Bryce 6d39d791b1 refactor: move safety to utils 5 months ago
Bryce 168a843f29 refactor: move colorize to api 5 months ago
Bryce 96f4268d44 refactor: move video_sample to api 5 months ago
Bryce e72e8992ab refactor: create api module 5 months ago
Bryce ad561e8833 refactor: move model_manager to utils 5 months ago
Bryce d478771cc0 refactor: move a bunch of stuff to utils 5 months ago
Bryce 987af23abe refactor: move train.py 5 months ago
Bryce 0c456cd52a refactor: remove lr_scheduler.py 5 months ago
Bryce 01e32ff3f6 refactor: move bin files 5 months ago
Bryce 316114e660 docs: add docstrings
Wrote an openai script and custom prompt to generate them.
5 months ago
jaydrennan e7b6fc40fa fix: adds default line ending for csv writing.
the csv library defaults to using CRLF line endings if not specified.
5 months ago
jaydrennan 3f3e080d39 feature: adds ability to use qrcode
feature: adds controlnet qrcode image generation.
feature: adds control net for qrcode image generation.
5 months ago