Add functions to export and import the model state for a given voice (#129)

2026-02-16 15:37:00 +01:00
parent aca7dc8db6
commit 8443de8e06
11 changed files with 123 additions and 212 deletions
@@ -11,7 +11,6 @@ def init_states(
    for module_name, module in model.named_modules():
        if not isinstance(module, StatefulModule):
            continue
-        module._module_absolute_name = module_name
        module_state = module.init_state(batch_size, sequence_length=sequence_length)
        result[module_name] = module_state
    return result