wehub-resource-sync
11 lines
655 B
Python
11 lines
655 B
Python
import os
|
|
|
|
# The rf-detr fast detector's DINOv2 pos-embed interpolation uses an antialiased-bicubic
|
|
# op that has no MPS kernel. PyTorch reads PYTORCH_ENABLE_MPS_FALLBACK at `import torch`,
|
|
# so set it here — before torch is imported — to let that one op fall back to CPU while
|
|
# the rest of the model runs on MPS. This is only effective when surya is imported before
|
|
# torch; when torch is already loaded, `_pick_device` (surya.common.rfdetr_torch) probes
|
|
# the op at runtime and falls back to CPU device selection instead. setdefault so we never
|
|
# clobber a value the user set deliberately.
|
|
os.environ.setdefault("PYTORCH_ENABLE_MPS_FALLBACK", "1")
|