Test

app.py

@@ -78,6 +78,7 @@ def response(state: AppState, audio: tuple):
                 tuple(torch.tensor(vec).cuda() for vec in tup)
                 for tup in state.model_outs.past_key_values
             )
+            print(outs)
     prev_outs = state.model_outs
     start = False
     for resp, outs in diva_audio(
@@ -95,8 +96,9 @@ def response(state: AppState, audio: tuple):
     del outs.hidden_states
     if spaces.config.Config.zero_gpu:
         outs.past_key_values = tuple(
-            tuple(vec.cpu().numpy for vec in tup) for tup in outs.past_key_values
+            tuple(vec.cpu().numpy() for vec in tup) for tup in outs.past_key_values
         )
+        print(outs)
     return (
         AppState(conversation=state.conversation),
         state.conversation,