imaginAIry/imaginairy/__init__.py

import os

# tells pytorch to allow MPS usage (for Mac M1 compatibility)
os.putenv("PYTORCH_ENABLE_MPS_FALLBACK", "1")

from .api import imagine, imagine_image_files  # noqa
from .schema import ImaginePrompt, ImagineResult, WeightedPrompt  # noqa
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			`import os`

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: face enhancement and upscaling!! 2022-09-13 07:27:53 +00:00			`from .api import imagine, imagine_image_files # noqa`
feature: tile mode from https://github.com/replicate/cog-stable-diffusion/compare/main...TomMoore515:material_stable_diffusion:main 2022-09-11 20:56:41 +00:00			`from .schema import ImaginePrompt, ImagineResult, WeightedPrompt # noqa`