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