spaces_compat.py

from __future__ import annotations

import os

IS_HF_SPACE = bool(os.getenv("SPACE_ID"))

if IS_HF_SPACE:
    import spaces as _spaces
    GPU = _spaces.GPU
else:
    def GPU(*args, **kwargs):
        def decorator(fn):
            return fn
        return decorator