This commit is contained in:
yumoqing 2024-09-09 18:02:54 +08:00
parent 0c186ad454
commit 49f0a82278
2 changed files with 5 additions and 0 deletions

View File

@ -2,6 +2,8 @@ import io
from mini_omni.inference import OmniInference from mini_omni.inference import OmniInference
from av import AudioFrame from av import AudioFrame
from aiortc import VideoStreamTrack, AudioStreamTrack
class LLMAudioStreamTrack(AudioStreamTrack): class LLMAudioStreamTrack(AudioStreamTrack):
def __init__(self, ckpt_dir): def __init__(self, ckpt_dir):
super().__init__() super().__init__()

View File

@ -1,3 +1,6 @@
import sys
sys.path.append('./mini_omni')
import asyncio import asyncio
import random import random
import json import json