Bryce
9cdacd454f
style: use latest ruff
2024-03-15 11:32:24 -07:00
Bryce
5b3b04b877
build: remove pytorch lightning dependency
2024-01-02 20:51:05 -08:00
jaydrennan
3322777f5e
refactor: formatting changes ruff formatter
...
ruff formatter has slight differences in formatting compared to black
2023-12-27 17:08:01 -08:00
jaydrennan
7eef3bf628
feature: replaces black formatter with ruff formatter
2023-12-27 17:08:01 -08:00
jaydrennan
e1e6f8037c
refactor: removes unused code and configurations ( #405 )
...
Co-authored-by: jaydrennan
2023-12-15 15:27:00 -08:00
Bryce
d478771cc0
refactor: move a bunch of stuff to utils
2023-12-15 14:32:01 -08:00
Bryce
316114e660
docs: add docstrings
...
Wrote an openai script and custom prompt to generate them.
2023-12-15 14:32:01 -08:00
Bryce
012cc648d3
style: fix all the mypy typing issues
...
...or ignore them.
2023-12-12 20:54:39 -08:00
Bryce
eae4f20ae2
ci: add type checker
...
fix some typehint issues
2023-12-12 20:54:39 -08:00
Bryce
558d3388e5
style: speed up linting and autoformatting. fix lints
2023-09-29 23:01:50 -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
54c3ad51d6
feature: controlnet
2023-02-22 23:38:47 -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
68e7fd73c5
perf: add xformers dependency
2023-02-15 14:22:00 -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
7bdde559cc
performance: memory management improvements ( #210 )
...
- tile mode made more efficient. especially when not being used
- add script to iteratively make bigger images
2023-01-26 03:14:02 -08:00
Bryce
248679d8de
feature: image sizes can now be multiples of 8 instead of 64
...
from https://github.com/CompVis/stable-diffusion/issues/60#issuecomment-1240294667
2023-01-23 22:31:25 -08:00
Bryce
7c05c3aae2
feature: image edit gifs and demo-reels
2023-01-21 17:50:31 -08:00
Bryce
2a3e19f5a1
feature: 🎉 Edit Images with Instructions alone!
2023-01-21 13:34:26 -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
1381c7fed4
lint: new ruff linter
2023-01-02 14:11:36 -08:00
Bryce
ad0b9e8ab8
feature: tile in a single dimension ("x" or "y")
2022-12-19 21:37:59 -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
4610d7f01d
feature: xformers support
...
add more upscaling code (that doesn't yet work)
2022-11-26 22:55:51 -08:00
Kian-Meng Ang
3d04df4dee
Fix typos
...
Found via `codespell -S ./imaginairy/vendored`
2022-11-25 19:23:06 -08:00
Bryce
58c2897dd1
refactor: fix lint issues
2022-11-25 14:39:20 -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
7af1ab66ca
fix: add workaround for bug in k_diffusion on mps
...
As documented here: https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/4558#issuecomment-1310387114
- make image logs more consistent
- note which step the progress images belong to in the filename
2022-11-12 20:19:23 -08:00
Bryce
021a0c540d
feature: inpainting model support; improved model manager
2022-10-25 20:07:35 -07:00
Bryce
4ba1965db8
feature: k-diff sampler img2img and masking
2022-10-14 03:23:16 -07:00
Bryce
e8bb3cf5fd
refactor: consolidate masking logic
2022-10-13 07:04:37 -07:00
Bryce
8d4b5cb9e1
refactor: standardize samplers more
2022-10-13 07:04:37 -07:00
Bryce
b0123a8f43
refactor: remove unused parameters
2022-10-11 01:06:24 -05:00
Bryce
e5c5df6b3d
tests: add some utils tests
2022-10-07 00:02:26 -05:00
Bryce
69af07ab67
style: fix lint issues
2022-09-24 00:31:08 -07:00
Bryce
a3a0de08e9
autoformat
2022-09-23 14:41:15 -07:00
Jay Drennan
df28bf8805
refactor: implements changes to comply with pylama
2022-09-22 12:52:37 -06:00
Bryce
ff7fc0dab7
feature: tile mode can now be specified per-prompt
2022-09-20 22:57:03 -07:00
Bryce
a46424c673
feature: img2img now supported with PLMS (instead of just DDIM)
...
Kinda hacky copy/pasting from ddim. Need to cleanup
2022-09-20 08:42:00 -07:00
Bryce
930295d840
feature: txt2mask - automated text replacement
...
from https://github.com/timojl/clipseg
2022-09-18 06:07:47 -07:00
Bryce
d7cbf6e416
refactor: cleanup ddim
...
- delete more unused code
- fix some lints
2022-09-17 14:02:27 -07:00
Bryce
6307a0daf5
refactor: cleanup ddim
2022-09-17 12:24:27 -07:00
Bryce
bb665b9eb6
fix: make k-diffusion samplers deterministic
...
- add test for hashes on mps. images look same on CUDA but are slightly different.
2022-09-14 09:37:45 -07:00
Bryce
b4a3b8c2b3
feature: k-diffusion samplers
...
- improved image logging functionality. can just stick log_latent wherever you want
- improved some variable naming
- moved all the samplers together
- vendored k-diffusion library
2022-09-14 00:40:25 -07:00
Bryce
ff7455034d
docs and lint
2022-09-13 00:46:37 -07:00
Bryce
541ecb9701
feature: face enhancement and upscaling!!
2022-09-13 00:27:53 -07:00
Bryce
6ac3d40009
docs: update readme. add docs to package
...
Clean up logging some more
2022-09-11 21:36:14 -07:00
Bryce
84a73cb5a2
feature: (wip) better image to image
...
I tried it with the DDIM sampler and it didn't work.
Probably need to use the k-diffusion sampler with it
from a846393251/find_noise.py
needs https://github.com/crowsonkb/k-diffusion
2022-09-11 18:00:40 -07:00
Bryce
438c2868ad
refactor: run import sorter
2022-09-11 13:58:14 -07:00