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 的列表。