Create update_config.py

update_config.py

@@ -0,0 +1,19 @@
+"""
+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()