Add code

inference.py

@@ -492,6 +492,7 @@ class OmniInference:
             if begin_generate:
                 current_index += 1
                 if current_index == nums_generate:
+                    print("yielding")
                     current_index = 0
                     snac = get_snac(list_output, index, nums_generate)
                     audio_stream = generate_audio_data(snac, self.snacmodel, self.device)