Archives/imaginAIry
2
0
Fork 0
mirror of https://github.com/brycedrennan/imaginAIry synced 2024-10-31 03:20:40 +00:00
Code Issues Packages Projects Releases Wiki Activity
360 Commits 10 Branches 104 Tags 322 MiB
38ac0b7f54
Commit Graph

97 Commits

Author SHA1 Message Date
Bryce
38ac0b7f54 tests: fix tests 2023-05-20 13:09:00 -07:00
Bryce
df25936d6f feature: automatic use of inpainting 
feature disabled since controlnet inpainting doesn't work great. Was disabled by setting `inpaint_method="finetune",`
 2023-05-20 13:09:00 -07:00
Bryce
d32e1060cd feature: multi-controlnet support at the command line 
add controlnet option for edit demo
 2023-05-20 13:09:00 -07:00
Bryce
bcaa000d35 fix: model logging 2023-05-20 13:09:00 -07:00
Bryce
926692ad03 tests: "prime" the controlnets 
Trying to get things working on m1. doesn't fix everything
 2023-05-20 13:09:00 -07:00
Bryce
eca97a25a0 tests: adjust tests to pass 2023-05-20 13:09:00 -07:00
Bryce
4c77fd376b feature: improvements to memory management 
not thoroughly tested on low-memory devices
 2023-05-20 13:09:00 -07:00
Bryce
6db296aa37 tests: faster tests 2023-05-20 11:35:49 -07:00
Bryce
c082ea523f tests: update tests 
- controlnet version changes + graphics card change
 2023-05-13 23:41:00 -07:00
Bryce
db8d3b74ea feature: add "detail" control mode. 
Add details to an image. Great for upscaling an image.
 2023-05-05 02:53:43 -07:00
Bryce
476a81a967 feature: add "shuffle" control mode 
Image is generated from elements of control image. similar to style transfer
 2023-05-05 02:53:43 -07:00
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
 2023-05-04 19:48:27 -07:00
Bryce
95d3d08d27 fix: better input image path handling 
throw exception for non-existent images
 2023-04-14 00:22:43 -07:00
Bryce
37d6642c83 fix: fix model downloads that were broken 
by [library change in transformers 4.27.0](8f3b4a1d5b)
 2023-03-18 13:49:11 -07:00
Bryce
3b777b98d8 fix: improved large images using composition 2023-02-28 21:04:00 -08:00
Bryce
40da2b0357 tests: fix tests 2023-02-28 21:04:00 -08:00
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
 2023-02-28 21:04:00 -08:00
Bryce
e3b23584ed refactor: split up command line tool 2023-02-25 14:29:55 -08:00
Bryce
b261c62d4e docs: update docs 2023-02-22 23:38:47 -08:00
Bryce
54c3ad51d6 feature: controlnet 2023-02-22 23:38:47 -08:00
Bryce
bcd761b8a6 tests: better coverage of cli commands 
add "empty" model that just uses empty state for models
 2023-02-20 11:06:54 -08:00
Bryce
7b684b6e5c test: add smoke test for edit command 2023-02-20 11:06:54 -08:00
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
 2023-02-18 15:36:34 -08:00
Bryce
b93b6a4d7c perf: use silu instead of nonlinearity for speedup 2023-02-15 14:22:00 -08:00
Bryce
003a512dc8 perf: mps perf improvement 
attention slicing wasn't working
 2023-02-15 14:22:00 -08:00
Bryce
ea1d4baafe refactor: cleanup image generation code 2023-02-15 14:22:00 -08:00
Bryce
8a97213622 feature: sliced latent decoding 
allows generation of bigger images. tile seams can be noticeable occasionally despite the feathering
 2023-02-12 01:41:57 -08:00
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.
 2023-02-02 21:43:04 -08:00
Bryce Drennan
9ee09ac842
feature: add compilation animations (#224) 
- add generation/compare gifs
 2023-01-28 17:16:47 -08:00
Bryce Drennan
542e4fbd55
fix: tile mode was broken since latest perf improvements (#220) 2023-01-27 22:56:46 -08:00
Bryce Drennan
7deabaae62
tests: update expected images for tests (#215) 2023-01-26 20:41:47 -08:00
Bryce
8855823293 fix: rename files for windows compat 2023-01-22 19:08:01 -08:00
Bryce
5b56f05da7 feature: print version 
- fix: default config wasn't being selected for custom models
 2023-01-18 12:04:07 -08:00
Bryce
81f294216b feature: 🎉 outpainting 2023-01-17 00:31:01 -08:00
Bryce
24e10f9e5f fix: don't report a safety issue when a black image is generated 2023-01-16 14:51:31 -08:00
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
 2023-01-16 01:35:59 -08:00
Bryce
9e0a9e2c29 tests: update 2023-01-02 14:11:36 -08:00
Bryce
eb40842078 test: more flexible embedding test 2023-01-02 14:11:36 -08:00
Bryce
1381c7fed4 lint: new ruff linter 2023-01-02 14:11:36 -08:00
Bryce
da0f1e1ee6 fix: make sure randomness is generated on cpu for consistency 2022-12-21 09:33:46 -08:00
Bryce
1f7403155e fix: more thorough cleaning of memory when switching models 
also cleanup up some test failures
 2022-12-18 22:00:29 -08:00
Bryce
f30823e0b5 feature: Stable Diffusion 2.1 2022-12-07 10:41:36 -08:00
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.
 2022-12-02 02:03:13 -08:00
Bryce
554fc1d0ec tests: update images 
- probably the k-diffusin update fixed some bugs
- add missing images
 2022-11-28 00:32:53 -08:00
Bryce
257752887d perf: add back memory efficiency improvements 
Removed these in a rush to get SD-2.0 out.
 2022-11-28 00:32:53 -08:00
Bryce
9c153a8d2d feature: support k-sampler methods for SD-2.0-v (768) 2022-11-26 16:18:08 -08:00
Bryce
015088507f feature/refactor/fix: better defaults. correct version in metadata 
- feature: use different default steps and image sizes depending on sampler and model selceted
- fix: #110 use proper version in image metadata
- refactor: samplers all have their own class that inherits from ImageSampler
 2022-11-26 16:18:08 -08:00
Kian-Meng Ang
3d04df4dee Fix typos 
Found via `codespell -S ./imaginairy/vendored`
 2022-11-25 19:23:06 -08:00
Bryce
e67341223b feature: Stable Diffusion 2.0 
working: CUDA and MacOS
working: 512p model with all samplers
working: inpainting with all samplers
working: 768p model with ddim sampler
 2022-11-24 00:50:57 -08:00
Bryce
893b041a8f feature: progress image callback 2022-11-15 18:06:03 -08:00