# 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")