mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
8 lines
255 B
Python
8 lines
255 B
Python
"""Pythonic AI generation of images and videos"""
|
|
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")
|