vall-e/vall_e/utils/__init__.py

from .utils import (
    dispatch_attribute,
    flatten_dict,
    gather_attribute,
    load_state_dict_non_strict,
    setup_logging,
    to_device,
    tree_map,
    do_gc,
    set_seed,
)