Commit Graph

8 Commits (master)

Author SHA1 Message Date
Bryce 1faea372f9 fix: cleanup logging - remove unnecessary version checks 2 months ago
jaydrennan 964dd4ead7 feature: integrates spandrel for upscaling 2 months ago
Bryce 76b6fa8b65 tests: fix them 3 months ago
Bryce a8acb451c5 ci: use `uv`
waiting for this issue to be resolved before using it for pip-compile

https://github.com/astral-sh/uv/issues/1624

and it didn't properly install the command line tools `aimg` and `imagine` so not using it for editable install on github either
3 months ago
Bryce 5b3b04b877 build: remove pytorch lightning dependency 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 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