diff --git a/library/device_utils.py b/library/device_utils.py index f6641ab5c..8823c5d9a 100644 --- a/library/device_utils.py +++ b/library/device_utils.py @@ -3,11 +3,6 @@ import torch -from .utils import setup_logging -setup_logging() -import logging -logger = logging.getLogger(__name__) - try: HAS_CUDA = torch.cuda.is_available() except Exception: @@ -64,7 +59,7 @@ def get_preferred_device() -> torch.device: device = torch.device("mps") else: device = torch.device("cpu") - logger.info(f"get_preferred_device() -> {device}") + print(f"get_preferred_device() -> {device}") return device @@ -82,8 +77,8 @@ def init_ipex(): is_initialized, error_message = ipex_init() if not is_initialized: - logger.error("failed to initialize ipex: {error_message}") + print("failed to initialize ipex:", error_message) else: return except Exception as e: - logger.error("failed to initialize ipex: {e}") + print("failed to initialize ipex:", e)