File size: 330 Bytes
afc85dc |
1 2 3 4 5 6 7 8 9 10 |
# sitecustomize.py (put at repo root)
import os
os.environ.setdefault(
"PYTORCH_CUDA_ALLOC_CONF",
"expandable_segments:True,max_split_size_mb:128",
)
# Optional knobs for our loader downscaling (safer defaults)
os.environ.setdefault("MATANYONE_MAX_EDGE", "640")
os.environ.setdefault("MATANYONE_TARGET_PIXELS", "400000")
|