imaginAIry/imaginairy/__init__.py

import os

# tells pytorch to allow MPS usage (for Mac M1 compatibility)
os.putenv("PYTORCH_ENABLE_MPS_FALLBACK", "1")
# use more memory than we should
os.putenv("PYTORCH_MPS_HIGH_WATERMARK_RATIO", "0.0")
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-03 05:43:04 +00:00			`import os`
feature: cleaned up logging - cleans up all the logging. hide most of it - create better readme. show example images - save metadata into image 2022-09-11 06:27:22 +00:00
refactor/test: logging suppression + hashed image test - simpler logging suppression for `transformers` library - suppress logging noise for running tests - get test running for all samplers on mps and cuda platforms - refactor safety model env variable to allow classification 2022-09-15 02:40:50 +00:00			`# tells pytorch to allow MPS usage (for Mac M1 compatibility)`
feature: cleaned up logging - cleans up all the logging. hide most of it - create better readme. show example images - save metadata into image 2022-09-11 06:27:22 +00:00			`os.putenv("PYTORCH_ENABLE_MPS_FALLBACK", "1")`
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 2023-11-16 03:46:56 +00:00			`# use more memory than we should`
			`os.putenv("PYTORCH_MPS_HIGH_WATERMARK_RATIO", "0.0")`