imaginAIry/imaginairy/__init__.py

7 lines
192 B
Python
Raw Normal View History

import os
os.putenv("PYTORCH_ENABLE_MPS_FALLBACK", "1")
2022-09-11 20:58:14 +00:00
from .api import imagine_image_files, imagine_images # noqa
from .schema import ImaginePrompt, ImagineResult, WeightedPrompt # noqa