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

Merge branch 'main' into dev

Browse Source
This commit is contained in:
Kohya S
2024-02-27 18:53:43 +09:00
parent 577e9913ca 2d7389185c
commit 074d32af20
2 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/typos.yml vendored
View File

@@ -18,4 +18,4 @@ jobs:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
- name: typos-action - name: typos-action
uses: crate-ci/typos@v1.16.26 uses: crate-ci/typos@v1.17.2

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)