imaginAIry/imaginairy/__init__.py

7 lines
185 B
Python
Raw Normal View History

import os
os.putenv("PYTORCH_ENABLE_MPS_FALLBACK", "1")
from .api import imagine, imagine_image_files # noqa
from .schema import ImaginePrompt, ImagineResult, WeightedPrompt # noqa