VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

revert logging #1137

Browse Source
This commit is contained in:
Kohya S
2024-02-25 10:43:14 +09:00
parent e0acb10f31
commit fccbee2727
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
library/device_utils.py
View File

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