23 lines
634 B
Python
23 lines
634 B
Python
# __init__.py
|
|
import os
|
|
|
|
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
|
|
# https://github.com/pytorch/pytorch/issues/27971#issuecomment-1768868068
|
|
os.environ["ONEDNN_PRIMITIVE_CACHE_CAPACITY"] = "1"
|
|
os.environ["LRU_CACHE_CAPACITY"] = "1"
|
|
# prevent CPU memory leak when run model on GPU
|
|
# https://github.com/pytorch/pytorch/issues/98688#issuecomment-1869288431
|
|
# https://github.com/pytorch/pytorch/issues/108334#issuecomment-1752763633
|
|
os.environ["TORCH_CUDNN_V8_API_LRU_CACHE_LIMIT"] = "1"
|
|
|
|
|
|
import warnings
|
|
|
|
warnings.simplefilter("ignore", UserWarning)
|
|
from iopaint.cli import typer_app
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
typer_app()
|