print log
Browse files
vietTTS/hifigan/mel2wave.py
CHANGED
|
@@ -39,5 +39,6 @@ def mel2wave(
|
|
| 39 |
wav = jnp.squeeze(wav)
|
| 40 |
print("wav : ")
|
| 41 |
print(wav)
|
|
|
|
| 42 |
audio = jax.device_get(wav)
|
| 43 |
return
|
|
|
|
| 39 |
wav = jnp.squeeze(wav)
|
| 40 |
print("wav : ")
|
| 41 |
print(wav)
|
| 42 |
+
jax.config.update('jax_platform_name', 'cpu')
|
| 43 |
audio = jax.device_get(wav)
|
| 44 |
return
|