WaveGru_TTS / extract_tacotrons_model.py
mrsu0994
upload f5-tts source
154f182
raw
history blame
216 Bytes
import pickle
import jax
dic = pickle.load(open("./mono_tts_cbhg_small_0700000.ckpt", "rb"))
del dic["optim_state_dict"]
dic = jax.device_get(dic)
pickle.dump(dic, open("./mono_tts_cbhg_small_0700000.ckpt", "wb"))