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

Replaced print with logger

Browse Source
This commit is contained in:
Yuta Hayashibe
2024-02-25 00:50:36 +09:00
parent e69d34103b
commit 5d5f39b6e6
5 changed files with 138 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
library/device_utils.py
View File

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