imaginAIry/imaginairy/__init__.py

import os.path

# 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 .enhancers.describe_image_blip import generate_caption  # noqa
from .schema import (  # noqa
    ImaginePrompt,
    ImagineResult,
    LazyLoadingImage,
    WeightedPrompt,
)

PKG_ROOT = os.path.dirname(__file__)
feature: automatic mask generation mask generation working but not integrated 2022-09-18 00:02:45 +00:00			`import os.path`
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: face enhancement and upscaling!! 2022-09-13 07:27:53 +00:00			`from .api import imagine, imagine_image_files # noqa`
feature: generate captions for images - add wip functionality for negative masks - ci: add code linter that removes unused imports - add instructions to install rust on osx 2022-09-20 04:15:38 +00:00			`from .enhancers.describe_image_blip import generate_caption # noqa`
feature: urls as init images - --init-image accepts urls - cleanup command line code 2022-09-16 06:06:59 +00:00			`from .schema import ( # noqa`
			`ImaginePrompt,`
			`ImagineResult,`
			`LazyLoadingImage,`
			`WeightedPrompt,`
			`)`
feature: automatic mask generation mask generation working but not integrated 2022-09-18 00:02:45 +00:00
			`PKG_ROOT = os.path.dirname(__file__)`