From 49f0a82278650a1f8e4bff299bfb5b892209bc3d Mon Sep 17 00:00:00 2001 From: yumoqing Date: Mon, 9 Sep 2024 18:02:54 +0800 Subject: [PATCH] bugfix --- rtcllm/a2a.py | 2 ++ rtcllm/rtc.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/rtcllm/a2a.py b/rtcllm/a2a.py index c93217c..1f6c47a 100644 --- a/rtcllm/a2a.py +++ b/rtcllm/a2a.py @@ -2,6 +2,8 @@ import io from mini_omni.inference import OmniInference from av import AudioFrame +from aiortc import VideoStreamTrack, AudioStreamTrack + class LLMAudioStreamTrack(AudioStreamTrack): def __init__(self, ckpt_dir): super().__init__() diff --git a/rtcllm/rtc.py b/rtcllm/rtc.py index 604d6cf..d83c7a7 100644 --- a/rtcllm/rtc.py +++ b/rtcllm/rtc.py @@ -1,3 +1,6 @@ +import sys +sys.path.append('./mini_omni') + import asyncio import random import json