imaginAIry/imaginairy/__init__.py

9 lines
256 B
Python
Raw Normal View History

"""Pythonic AI generation of images and videos"""
2024-03-15 18:26:12 +00:00
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")