Commit Graph

39 Commits

Author SHA1 Message Date
Robin Rombach
f2fa96b7e5 README updates for SDXL 1.0 release (#68)
* remove sdxl report from github repo, point to arxiv instead

* update licenses and add teaser img

* update readme for SDXL 1.0 release
2023-07-26 19:24:52 +02:00
Aarni Koskela
c60c091f4d Move CODEOWNERS so it has an effect (#66)
* Move CODEOWNERS so it has an effect

* CI: use vars.SGM_CHECKPOINTS_PATH
2023-07-26 08:52:59 -07: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
Jonas Müller
4a3f0f546e Revert "Replace most print()s with logging calls (#42)" (#65)
This reverts commit 6f6d3f8716.
2023-07-26 10:30:21 +02:00
Tim Dockhorn
7934245835 Revert "Minimize re-exports from __init__ files (#44)" (#63)
This reverts commit 57862fb4c7.
2023-07-26 10:26:28 +02:00
Tim Dockhorn
1da250906d Revert "Dead code removal (#48)" (#62)
This reverts commit b5b5680150.
2023-07-26 10:26:00 +02:00
Tim Dockhorn
a4ceca6d03 Revert "fall back to vanilla if xformers is not available (#51)" (#61)
This reverts commit ef520df1db.
2023-07-26 10:25:17 +02:00
ablattmann
68f3f89bd3 Fix crashing line in logging in sgm/models/diffusion.py (#64) 2023-07-26 10:13:42 +02:00
Aarni Koskela
57862fb4c7 Minimize re-exports from __init__ files (#44)
This allows importing parts of the package without having to
import practically everything (since importing a package will
import its parents' __init__s, etc).
2023-07-25 16:24:09 +02:00
Aarni Koskela
ef520df1db fall back to vanilla if xformers is not available (#51) 2023-07-25 16:21:51 +02:00
Aarni Koskela
2897fdc99a Move attention testing/benchmarking code out of package (#47) 2023-07-25 15:40:22 +02:00
Aarni Koskela
b5b5680150 Dead code removal (#48)
* Remove old commented-out attention code

* Mark two functions as likely unused

* Use exists() and default() from sgm.util
2023-07-25 15:24:24 +02:00
Aarni Koskela
6f6d3f8716 Replace most print()s with logging calls (#42) 2023-07-25 15:21:30 +02:00
Luca Antiga
6ecd0a900a Fix link (#24) 2023-07-25 09:58:18 +02:00
Jonas Müller
e25e4c0df1 Merge pull request #43 from akx/fix-safetensors-load
Fix loading safetensors with load_model_from_config
2023-07-21 18:00:53 +02:00
Aarni Koskela
e5dc9669ed Set up Python packaging (#17)
* Sort .gitignore; add dist and *.py[cod]

* Use pyproject.toml + Hatch instead of setup.py

Sibling of https://github.com/Stability-AI/stablediffusion/pull/269

* Add packaging documentation
2023-07-18 13:06:05 +02:00
Aarni Koskela
48904a692d Fix loading safetensors with load_model_from_config
Previously, the `sd` from the safetensors if branch was not used at all, and `pl_sd` would have not been assigned.
2023-07-17 09:56:35 +03:00
Tim
5c10deee76 Merge branch 'main' of https://github.com/Stability-AI/generative-models into main 2023-07-09 10:40:26 -07:00
Tim
89f5413e6d Getting rid of unnecessary error message 2023-07-09 10:40:18 -07:00
Tim
ba3e7fed5a Fixing additional GPU memory on device 0 due to discretization 2023-07-09 10:40:09 -07:00
Tim Dockhorn
ea89ce793d Merge pull request #28 from jenuk/fix-samples_z
Fix `samples_z` undefined
2023-07-09 10:35:04 -07:00
jenuk
7b1978e055 Only do refiner step if samples are actually available 2023-07-07 07:48:21 +00:00
Jonas Müller
9d5ace911e Merge pull request #25 from pharmapsychotic/bugfix/watermark
Fix channel ordering RGB to cv2 BGR
2023-07-06 16:12:40 +02: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
Robin Rombach
ae18ba3e87 add sdxl report 2023-07-04 13:21:13 +02:00
Tim
061d11d55d Fixing validation step for PL 1 2023-06-30 13:40:24 -07:00
Tim
2796c81a5f Making spacing function slimmer 2023-06-30 11:17:57 -07:00
Tim
e9869d7822 Changed LegacyDDPMDiscretization for sampling 2023-06-30 11:17:46 -07:00
Tim Dockhorn
613af104c6 Merge pull request #14 from patrickvonplaten/patch-1
Update sampling.py
2023-06-29 16:17:03 -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
Ikko Eltociear Ashimine
5f0a2fcf48 Update README.md (#9) 2023-06-26 09:40:12 +02:00
Balanagireddy M
d8a6a97fb0 Minor spellings (#12) 2023-06-26 09:38:55 +02:00
Tim
a1af4ac4f1 Adapting txt logging for python 3.8 2023-06-25 16:42:45 -07:00
Robin Rombach
58ddbee3ee Update README.md 2023-06-22 10:48:49 -07:00
Robin Rombach
bec98beff8 Update README.md 2023-06-22 10:48:18 -07:00
Andreas Blattmann
081e0d4629 soon is now 0.0.1 2023-06-22 09:53:12 -07:00