update_config.py

"""
Here, we override config to have num_workers=0 because
of a limitation in HF Spaces Docker /dev/shm.
"""

import json
from pathlib import Path


def main(config_file="configs/44k/config.json"):
    config_path = Path(config_file)
    data = json.loads(config_path.read_text())
    data['train']['batch_size'] = 16
    data['train']['eval_interval'] = 800
    data['train']['num_workers'] = 0
    data['train']['persistent_workers'] = False
    data['train']['push_to_hub'] = True
    data['train']['repo_id'] = tuple(data['spk'])[0]
    data['train']['private'] = True
    config_path.write_text(json.dumps(data, indent=2, sort_keys=False))

if __name__ == "__main__":
    main()