Commit Graph

102 Commits (07b097e001a3eb9c47828a51639403de49a2ece4)

Author SHA1 Message Date
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
11 months ago
Bryce 558d3388e5 style: speed up linting and autoformatting. fix lints 1 year ago
Bryce 8243ed616d fix: pydantic models for http server working now. Fixes #380 1 year ago
Bryce 7c2004bfcc feature/fix: migrate to pydantic 2.3
- test: add schema tests/fuzzer and fixes
 - fix default prompt. add tests
 - fix outpaint and controlnet defaults
 - fix init image strength defaults
1 year ago
Bryce 82d74c6b49 feature: switch to pydantic models
- allow prompt re-use by deferring random seed
1 year ago
Bryce 38ac0b7f54 tests: fix tests 1 year ago
Bryce df25936d6f feature: automatic use of inpainting
feature disabled since controlnet inpainting doesn't work great. Was disabled by setting `inpaint_method="finetune",`
1 year ago
Bryce d32e1060cd feature: multi-controlnet support at the command line
add controlnet option for edit demo
1 year ago
Bryce bcaa000d35 fix: model logging 1 year ago
Bryce 926692ad03 tests: "prime" the controlnets
Trying to get things working on m1. doesn't fix everything
1 year ago
Bryce eca97a25a0 tests: adjust tests to pass 1 year ago
Bryce 4c77fd376b feature: improvements to memory management
not thoroughly tested on low-memory devices
1 year ago
Bryce 6db296aa37 tests: faster tests 1 year ago
Bryce c082ea523f tests: update tests
- controlnet version changes + graphics card change
1 year ago
Bryce db8d3b74ea feature: add "detail" control mode.
Add details to an image. Great for upscaling an image.
1 year ago
Bryce 476a81a967 feature: add "shuffle" control mode
Image is generated from elements of control image. similar to style transfer
1 year ago
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
1 year ago
Bryce 95d3d08d27 fix: better input image path handling
throw exception for non-existent images
2 years ago
Bryce 37d6642c83 fix: fix model downloads that were broken
by [library change in transformers 4.27.0](8f3b4a1d5b)
2 years ago
Bryce 3b777b98d8 fix: improved large images using composition 2 years ago
Bryce 40da2b0357 tests: fix tests 2 years ago
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
2 years ago
Bryce e3b23584ed refactor: split up command line tool 2 years ago
Bryce b261c62d4e docs: update docs 2 years ago
Bryce 54c3ad51d6 feature: controlnet 2 years ago
Bryce bcd761b8a6 tests: better coverage of cli commands
add "empty" model that just uses empty state for models
2 years ago
Bryce 7b684b6e5c test: add smoke test for edit command 2 years ago
Bryce 1563e0b871 test: add some autoencoder tests
the fold-unfold encoding/decoding looks like it's slower and has worse seams than the sliced feathering approach
2 years ago
Bryce b93b6a4d7c perf: use silu instead of nonlinearity for speedup 2 years ago
Bryce 003a512dc8 perf: mps perf improvement
attention slicing wasn't working
2 years ago
Bryce ea1d4baafe refactor: cleanup image generation code 2 years ago
Bryce 8a97213622 feature: sliced latent decoding
allows generation of bigger images. tile seams can be noticeable occasionally despite the feathering
2 years ago
Bryce Drennan 9eacf5e7ed
perf: improve startup time by doing some imports lazily (#233)
just running `aimg --help` or `aimg --version` was very slow due to all the imports being brought in eagerly

Before changes `aimg --help`
`2.24s user 4.05s system 184% cpu 3.416 total`

After changes:
`0.04s user 0.02s system 8% cpu 0.625 total`

Used `PYTHONPROFILEIMPORTTIME=1 aimg --help` to find time consuming imports.

Also switched to using `scripts` instead of `entrypoints` since the scripts are much faster.

Made duplicate SAMPLER_TYPE_OPTIONS that can be loaded without loading all the samplers themselves.

Likely a breaking change - not sure.
2 years ago
Bryce Drennan 9ee09ac842
feature: add compilation animations (#224)
- add generation/compare gifs
2 years ago
Bryce Drennan 542e4fbd55
fix: tile mode was broken since latest perf improvements (#220) 2 years ago
Bryce Drennan 7deabaae62
tests: update expected images for tests (#215) 2 years ago
Bryce 8855823293 fix: rename files for windows compat 2 years ago
Bryce 5b56f05da7 feature: print version
- fix: default config wasn't being selected for custom models
2 years ago
Bryce 81f294216b feature: 🎉 outpainting 2 years ago
Bryce 24e10f9e5f fix: don't report a safety issue when a black image is generated 2 years ago
Bryce 5cc73f6087 feature: finetuning
- feature: finetuning your own image models
- feature: image prep command. crops to face or other interesting parts of photo
- fix: back-compat for hf_hub_download
- feature: add prune-ckpt command
- feature: allow specification of model config file
2 years ago
Bryce 9e0a9e2c29 tests: update 2 years ago
Bryce eb40842078 test: more flexible embedding test 2 years ago
Bryce 1381c7fed4 lint: new ruff linter 2 years ago
Bryce da0f1e1ee6 fix: make sure randomness is generated on cpu for consistency 2 years ago
Bryce 1f7403155e fix: more thorough cleaning of memory when switching models
also cleanup up some test failures
2 years ago
Bryce f30823e0b5 feature: Stable Diffusion 2.1 2 years ago
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.
2 years ago
Bryce 554fc1d0ec tests: update images
- probably the k-diffusin update fixed some bugs
- add missing images
2 years ago
Bryce 257752887d perf: add back memory efficiency improvements
Removed these in a rush to get SD-2.0 out.
2 years ago