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']['num_workers'] = 0
    data['train']['persistent_workers'] = False
    config_path.write_text(json.dumps(data, indent=2, sort_keys=False))


if __name__ == "__main__":
    main()