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