跳到主要内容

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/20logging.DEBUG/10),用于控制详细程度。

返回值:

  • None

list_models

def list_models()

列出可用的模型 ID。

返回值:

  • list - 一个包含可用模型 ID 的列表。