utils
AlphaCubeパッケージのユーティリティ。
このモジュールは、AlphaCubeパッケージ全体で使用される共通のユーティリティを提供します。これには以下が含まれます:
- スタイル付きコンソール出力のための事前設定済み
rich
ロガー。 - 利用可能な最適な
torch.device
(CUDA、MPS、またはCPU)と対応するtorch.dtype
の自動検出。 - 詳細度を制御するユーティリティ関数(
set_verbose
)、利用可能なモデルを一覧表示する関数(list_models
)、およびグローバルキャッシュディレクトリを定義する関数。
set_verbose
def set_verbose(loglevel=20)
ロガーの詳細度レベルを設定します。
引数:
loglevel
int - 詳細度を制御するためのロギングレベル(例:logging.INFO
/20
、logging.DEBUG
/10
)。
戻り値:
None
list_models
def list_models()
利用可能なモデルIDを一覧表示します。
戻り値:
list
- 利用可能なモデルIDのリスト。