fix prompt not on cuda

core/bark/generate_audio.py

@@ -38,8 +38,17 @@ def generate_audio(
         generation_config: configurations to generate audio
 
     """
+
+    if prompt is not None:
+        if prompt.semantic_prompt is not None:
+            prompt.semantic_prompt = prompt.semantic_prompt.to("cuda")
+        if prompt.coarse_prompt is not None:
+            prompt.coarse_prompt = prompt.coarse_prompt.to("cuda")
+        if prompt.fine_prompt is not None:
+            prompt.fine_prompt = prompt.fine_prompt.to("cuda")
+
     if prompt is not None:
-        semantic_prompt = prompt.semantic_prompt.to("cuda") if prompt is not None else None
+        semantic_prompt = prompt.semantic_prompt if prompt is not None else None
         # if len(semantic_prompt.shape) == 2:
         #     semantic_prompt = semantic_prompt[0, :]
         assert (