diff --git a/config/chat_with_qwen_omni.yaml b/config/chat_with_qwen_omni.yaml index 7d8bb1fe..6ae76bda 100644 --- a/config/chat_with_qwen_omni.yaml +++ b/config/chat_with_qwen_omni.yaml @@ -16,7 +16,7 @@ default: module: client/rtc_client/client_handler_rtc connection_ttl: 36000 # max time a session will last for - connection_ttl: 900 + # connection_ttl: 900 SileroVad: module: vad/silerovad/vad_handler_silero speaking_threshold: 0.5 @@ -52,10 +52,16 @@ default: # force_create_avatar: false # Whether to force regenerate digital human data # debug: false # Whether to enable debug mode # multi_thread_inference: true # true: use 2 threads to inference (unet, vae), false: use single-thread to inference + # LiteAvatar: + # module: avatar/liteavatar/avatar_handler_liteavatar + # avatar_name: ~/.cache/modelscope/hub/models/HumanAIGC-Engineering/LiteAvatarGallery/20250408/sample_data/ + # fps: 25 + # debug: false + # enable_fast_mode: false + # use_gpu: false + LiteAvatar: module: avatar/liteavatar/avatar_handler_liteavatar avatar_name: 20250408/sample_data fps: 25 - debug: false - enable_fast_mode: false - use_gpu: true + use_gpu: false \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index 2001bbb6..ae99507f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,9 +28,7 @@ dependencies = [ "tqdm~=4.67.1", "typing-extensions~=4.12.2", "uvicorn~=0.34.0", - "torch==2.8.0", "torchvision", - "torchaudio", "python-dotenv>=1.1.0", ] diff --git a/src/handlers/avatar/liteavatar/pyproject.toml b/src/handlers/avatar/liteavatar/pyproject.toml index a8645d71..9d5f7e51 100644 --- a/src/handlers/avatar/liteavatar/pyproject.toml +++ b/src/handlers/avatar/liteavatar/pyproject.toml @@ -13,5 +13,5 @@ dependencies = [ "typeguard==2.13.3", "vector-quantize-pytorch~=1.21.7", "vocos~=0.1.0", - "onnxruntime-gpu~=1.20.1", + "onnxruntime~=1.20.1", ] diff --git a/src/handlers/avatar/liteavatar/shared_memory_buffer_pool.py b/src/handlers/avatar/liteavatar/shared_memory_buffer_pool.py index 5cf40978..7a6fd06c 100644 --- a/src/handlers/avatar/liteavatar/shared_memory_buffer_pool.py +++ b/src/handlers/avatar/liteavatar/shared_memory_buffer_pool.py @@ -160,7 +160,8 @@ def _attach_audio_pool(self, shm_names: List[str]): except Exception as e: logger.error(f"Failed to attach to audio buffer {i} ({shm_name}): {e}") self.audio_buffers.append(None) - logger.info(f"Audio attached: {len(self.audio_buffers)} buffers, queue: {self.audio_available.qsize()}") + qsize = self._safe_qsize(self.audio_available) + logger.info(f"Audio attached: {len(self.audio_buffers)} buffers, queue: {qsize}") def _attach_video_pool(self, shm_names: List[str]): """Attach to existing video buffer pool.""" @@ -428,4 +429,3 @@ def __del__(self): self.cleanup() except Exception as e: logger.error(f"Error in __del__: {e}") -