Commit Graph

38 Commits

Author SHA1 Message Date
Stephan Auerhahn
2fc4680bf9 Easier default params 2023-08-12 13:22:04 -07:00
Stephan Auerhahn
e32972b85b remove extra init 2023-08-12 05:42:22 -07:00
Stephan Auerhahn
65c6ec1cec run black 2023-08-12 05:40:25 -07:00
Stephan Auerhahn
fbe93fc53b PR fixes, model specific defaults 2023-08-12 05:33:16 -07:00
Stephan Auerhahn
c0655731d5 fix streamlit inputs 2023-08-12 04:25:56 -07:00
Stephan Auerhahn
fe4632034b fix for orig dimensions 2023-08-11 16:31:53 -07:00
Stephan Auerhahn
d6f2b78994 pass options into state2 init 2023-08-10 15:06:55 -07:00
Stephan Auerhahn
cd81956241 text updates 2023-08-10 13:31:03 -07:00
Stephan Auerhahn
5c17043434 change default 2023-08-10 13:15:23 -07:00
Stephan Auerhahn
2aebc8882d split fp16 and swapping functionality 2023-08-10 13:14:38 -07:00
Stephan Auerhahn
3816aaa639 simplify device_manager usage 2023-08-10 13:05:30 -07:00
Stephan Auerhahn
88395261d8 update helpers 2023-08-10 12:45:37 -07:00
Stephan Auerhahn
b3866d1218 move checkbox out of cached resource 2023-08-10 12:44:48 -07:00
Stephan Auerhahn
a25662e969 low vram checkbox fix, remove magic strings 2023-08-10 12:40:32 -07:00
Stephan Auerhahn
26b10f56f3 fix missing index 2023-08-10 12:24:12 -07:00
Stephan Auerhahn
de7a627978 more fixes and cleanup 2023-08-10 05:11:34 -07:00
Stephan Auerhahn
47805f233c finish device manager refactor 2023-08-10 04:55:43 -07:00
Stephan Auerhahn
e190ecc60b path helper & model swapping rewrite 2023-08-10 04:35:59 -07:00
Stephan Auerhahn
fc498bfaef remove duplicate imports 2023-08-10 03:20:56 -07:00
Stephan Auerhahn
8011d54ca1 some PR fixes 2023-08-10 03:19:37 -07:00
Stephan Auerhahn
b51c36b0df extract path resolution method, fix/improve device swapping support 2023-08-09 19:31:59 -07:00
Stephan Auerhahn
725bea9f75 pull in import fix 2023-08-09 13:29:16 -07:00
Stephan Auerhahn
a009aa8a9f adding some typing 2023-08-09 13:27:30 -07:00
Stephan Auerhahn
f86ffac274 context manager 2023-08-09 12:38:44 -07:00
Stephan Auerhahn
a726ce3eb7 replace usage of get 2023-08-09 12:30:43 -07:00
Stephan Auerhahn
c4b7baf896 Streamlit refactor (#105)
* initial streamlit refactoring pass

* cleanup and fixes

* fix refiner strength

* Modify params correctly

* fix exception
2023-08-06 19:58:52 -07:00
Stephan Auerhahn
f06c67c206 formatting, remove reference 2023-08-06 11:30:40 +00:00
Stephan Auerhahn
b216934b7e align with streamlit helpers and re-de-deuplicate 2023-08-06 11:20:22 +00:00
Jonas Müller
e5d714d304 Improved sampling (#69)
* New research features

* Add new model specs
---------

Co-authored-by: Dominik Lorenz <53151171+qp-qp@users.noreply.github.com>

* remove sd1.5 and change default refiner to 1.0

* remove asking second time for output

* adapt model names

* adjusted strength

* Correctly pass prompt

---------

Co-authored-by: Dominik Lorenz <53151171+qp-qp@users.noreply.github.com>
2023-07-26 19:49:23 +02:00
Stephan Auerhahn
931d7a389a Add inference helpers & tests (#57)
* Add inference helpers & tests

* Support testing with hatch

* fixes to hatch script

* add inference test action

* change workflow trigger

* widen trigger to test

* revert changes to workflow triggers

* Install local python in action

* Trigger on push again

* fix python version

* add CODEOWNERS and change triggers

* Report tests results

* update action versions

* format

* Fix typo and add refiner helper

* use a shared path loaded from a secret for checkpoints source

* typo fix

* Use device from input and remove duplicated code

* PR feedback

* fix call to load_model_from_config

* Move model to gpu

* Refactor helpers

* cleanup

* test refiner, prep for 1.0, align with metadata

* fix paths on second load

* deduplicate streamlit code

* filenames

* fixes

* add pydantic to requirements

* fix usage of `msg` in demo script

* remove double text

* run black

* fix streamlit sampling when returning latents

* extract function for streamlit output

* another fix for streamlit outputs

* fix img2img in streamlit

* Make fp16 optional and fix device param

* PR feedback

* fix dict cast for dataclass

* run black, update ci script

* cache pip dependencies on hosted runners, remove extra runs

* install package in ci env

* fix cache path

* PR cleanup

* one more cleanup

* don't cache, it filled up
2023-07-26 04:37:24 -07:00
Benjamin Aubin
e596332148 Pre release changes for production (#59)
* clean requirements

* rm taming deps

* isort, black

* mv lipips, license

* clean vq, fix path

* fix loss path, gitignore

* tested requirements pt13

* fix numpy req for python3.8, add tests

* fix name

* fix dep scipy 3.8 pt2

* add black test formatter
2023-07-26 12:09:28 +02:00
jenuk
7b1978e055 Only do refiner step if samples are actually available 2023-07-07 07:48:21 +00:00
pharmapsychotic
95b9acc5c6 Reformat with black 2023-07-06 09:03:23 -05:00
pharmapsychotic
5df4d9893c Watermark encoder expects images in BGR channel order (matching cv2 imread). This fix reduces the watermark artifacts. 2023-07-05 12:05:14 -05:00
Tim
e9869d7822 Changed LegacyDDPMDiscretization for sampling 2023-06-30 11:17:46 -07:00
Patrick von Platen
376cec3b0f Update sampling.py
Correct typo
2023-06-26 14:37:10 +02:00
Bryce Drennan
76e549dd94 Add missing init files (#11) 2023-06-26 09:41:25 +02:00
Andreas Blattmann
081e0d4629 soon is now 2023-06-22 09:53:12 -07:00