Skip to main content

utils

Utilities for the AlphaCube Package.

This module provides common utilities used across the AlphaCube package, including:

  • A pre-configured rich logger for styled console output.
  • Automatic detection of the best available torch.device (CUDA, MPS, or CPU) and the corresponding torch.dtype.
  • Utility functions for controlling verbosity (set_verbose), listing available models (list_models), and defining the global cache directory.

set_verbose​

def set_verbose(loglevel=20)

Set the verbosity level of the logger.

Arguments:

  • loglevel int - Logging level (e.g., logging.INFO/20, logging.DEBUG/10) to control the verbosity.

Returns:

  • None

list_models​

def list_models()

List the available model IDs.

Returns:

  • list - A list of available model IDs.