Spaces:

rahul7star
/

Magi

Paused

Magi / patch_configs.py

Create patch_configs.py

f38384a verified 7 months ago

390 Bytes

import json, glob

BASE = "/tmp/models"

for cfg in glob.glob("/workspace/MagiCompiler/example/*/config.json"):
    with open(cfg) as f:
        data = json.load(f)

    data["text_encoder_path"] = f"{BASE}/t5"
    data["audio_model_path"] = f"{BASE}/audio"
    data["vae_path"] = f"{BASE}/vae"

    with open(cfg, "w") as f:
        json.dump(data, f, indent=2)

    print("Patched:", cfg)

	import json, glob

	BASE = "/tmp/models"

	for cfg in glob.glob("/workspace/MagiCompiler/example/*/config.json"):
	with open(cfg) as f:
	data = json.load(f)

	data["text_encoder_path"] = f"{BASE}/t5"
	data["audio_model_path"] = f"{BASE}/audio"
	data["vae_path"] = f"{BASE}/vae"

	with open(cfg, "w") as f:
	json.dump(data, f, indent=2)

	print("Patched:", cfg)