fix speedyspeech inference input type

PaddlePaddle · Oct 27, 2021 · 041a18b · 041a18b
1 parent c2dd149
commit 041a18b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/speedyspeech/baker/inference.py b/examples/speedyspeech/baker/inference.py
@@ -98,8 +98,8 @@ def main():
             sentence, merge_sentences=True, get_tone_ids=True)
         phone_ids = input_ids["phone_ids"]
         tone_ids = input_ids["tone_ids"]
-        phones = phone_ids[0]
-        tones = tone_ids[0]
+        phones = phone_ids[0].numpy()
+        tones = tone_ids[0].numpy()
 
         if args.enable_auto_log:
             logger.times.stamp()