File size: 541 Bytes
5ebfade | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | import os
import logging
logger = logging.getLogger("dinov2")
def _xformers_is_available(layer):
XFORMERS_ENABLED = os.environ.get("XFORMERS_DISABLED") is None
xformers = None
try:
if XFORMERS_ENABLED:
import xformers
logger.info(f"xFormers is available ({layer})")
else:
logger.warning(f"xFormers is disabled ({layer})")
raise ImportError
except ImportError:
logger.warning(f"xFormers is not available ({layer})")
return xformers is not None
|