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,
    passes_policy,
    get_devices,
    truncate_json,
    timer,
    prune_missing,
    clamp,
    md5_hash,
    convert_kwargs
)
Rewrite init 2023-08-02 21:53:35 +00:00			`from .utils import (`
			`dispatch_attribute,`
			`flatten_dict,`
			`gather_attribute,`
			`load_state_dict_non_strict,`
			`setup_logging,`
			`to_device,`
			`tree_map,`
			`do_gc,`
backport fix from tortoise_tts with local trainer + loading state when training lora 2024-06-25 18:41:29 +00:00			`set_seed,`
fix issue with sft and shared tensors... 2024-08-05 00:56:21 +00:00			`passes_policy,`
oops 2024-09-17 20:25:12 +00:00			`get_devices,`
sped up inferencing by not doing .tolist() for rep pen / length pen (and a bug fix in the web UI from prev commit) 2024-10-05 03:18:20 +00:00			`truncate_json,`
cleaned up unused config flags, allow less strict yaml by pruning missing keys, renamed some dataset configs to be more unified 2024-10-17 22:06:48 +00:00			`timer,`
all I can do now until I wait for the model to (re)train for pure NAR 2024-11-10 04:57:34 +00:00			`prune_missing,`
use homwbrewed caching system for dataloader paths / durations (I'm pretty sure I am now triggering OOM killers with my entire dataset used) 2024-11-11 22:32:08 +00:00			`clamp,`
overhauled inference/sampler kwargs to stop being a bloated mess 2024-11-12 02:21:16 +00:00			`md5_hash,`
			`convert_kwargs`
Rewrite init 2023-08-02 21:53:35 +00:00			`)`