mirror of
https://github.com/unslothai/unsloth.git
synced 2026-08-07 07:54:10 +00:00
6d8c18cd1a
* Replace standalone Studio wording with Unsloth Replace the single word Studio with Unsloth wherever it is used as shorthand for Unsloth Studio in docs, CLI output, UI strings, i18n locales, workflow display names, comments and docstrings. Kept unchanged: the full name Unsloth Studio, third party product names (LM Studio, Visual Studio, Mac Studio), feature names (Recipe Studio, Fine-tuning Studio and its translations), and all identifiers such as env vars, commands, paths and filenames. * Address review feedback on the Studio wording rename Use "an" before Unsloth where the rename left the article as "a". Restore the split brand where Unsloth and Studio render as two halves of the full product name: the onboarding sidebar subtitle and the IPv6 localhost warning. Scope two messages to the full name Unsloth Studio where plain Unsloth was misleading: the AMD README bullet and the CLI studio setup error.
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""Early CPU thread-pool configuration for Unsloth processes."""
|
|
|
|
import os
|
|
from typing import MutableMapping, Optional
|
|
|
|
|
|
_THREAD_POOL_ENV_VARS = (
|
|
"OMP_NUM_THREADS",
|
|
"MKL_NUM_THREADS",
|
|
"OPENBLAS_NUM_THREADS",
|
|
"NUMEXPR_NUM_THREADS",
|
|
)
|
|
|
|
|
|
def configure_cpu_threads(env: Optional[MutableMapping[str, str]] = None) -> None:
|
|
"""Apply ``UNSLOTH_CPU_THREADS`` to native CPU pools when configured.
|
|
|
|
Must run before importing libraries that initialize an OpenMP or BLAS
|
|
pool. Library-specific vars are left untouched so users can override a
|
|
single runtime independently.
|
|
"""
|
|
environ = os.environ if env is None else env
|
|
configured = environ.get("UNSLOTH_CPU_THREADS", "").strip()
|
|
if not configured:
|
|
return
|
|
|
|
try:
|
|
thread_count = int(configured)
|
|
except ValueError as exc:
|
|
raise ValueError("UNSLOTH_CPU_THREADS must be a positive integer") from exc
|
|
if thread_count < 1:
|
|
raise ValueError("UNSLOTH_CPU_THREADS must be a positive integer")
|
|
|
|
value = str(thread_count)
|
|
for variable in _THREAD_POOL_ENV_VARS:
|
|
environ.setdefault(variable, value)
|