Merge pull request #429 from bluenote10:bugfix/fix_slow_data_preparation

Magenta Team · Magenta Team · commit b9b68b395a6b · 2022-03-24T15:49:55.000-07:00
PiperOrigin-RevId: 437105634
diff --git a/ddsp/training/data_preparation/prepare_tfrecord_lib.py b/ddsp/training/data_preparation/prepare_tfrecord_lib.py
@@ -69,7 +69,7 @@ def _chunk_audio(ex, sample_rate, chunk_secs):
   chunks = tf.signal.frame(audio, chunk_size, chunk_size, pad_end=True)
   n_chunks = chunks.shape[0]
   for i in range(n_chunks):
-    yield {'audio': chunks[i]}
+    yield {'audio': chunks[i].numpy()}
 
 
 def _add_f0_estimate(ex, frame_rate, center, viterbi):
diff --git a/ddsp/version.py b/ddsp/version.py
@@ -19,4 +19,4 @@
 pulling in all the dependencies in __init__.py.
 """
 
-__version__ = '3.2.3'
+__version__ = '3.3.0'
