deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-05 02:09:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

update demo port

Browse Source
This commit is contained in:
Hongji Zhu
2025-01-14 23:57:57 +08:00
parent f0d641b578
commit 3c2f729530
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
requirements_o2.6.txt
View File

@@ -13,6 +13,8 @@ librosa==0.9.0
decord decord
moviepy moviepy
# for web # for web demo
aiofiles==24.1.0
onnxruntime==1.20.1
fastapi fastapi
uvicorn uvicorn

6
web_demos/minicpm-o_2.6/model_server.py
View File

@@ -54,7 +54,7 @@ app = FastAPI()
logger = setup_logger() logger = setup_logger()
ap = argparse.ArgumentParser() ap = argparse.ArgumentParser()
ap.add_argument('--port', type=int , default=8088) ap.add_argument('--port', type=int , default=32550)
args = ap.parse_args() args = ap.parse_args()
@@ -523,13 +523,13 @@ class StreamManager:
for r in self.minicpmo_model.streaming_generate( for r in self.minicpmo_model.streaming_generate(
session_id=str(self.session_id), session_id=str(self.session_id),
tokenizer=self.minicpmo_tokenizer, tokenizer=self.minicpmo_tokenizer,
use_tts=True, generate_audio=True,
# enable_regenerate=True, # enable_regenerate=True,
): ):
if self.stop_response: if self.stop_response:
self.generate_end() self.generate_end()
return return
audio_np, sr, text = r audio_np, sr, text = r["audio_wav"], r["sampling_rate"], r["text"]
output_audio_path = self.savedir + f'/output_audio_log/output_audio_{self.output_audio_id}.wav' output_audio_path = self.savedir + f'/output_audio_log/output_audio_{self.output_audio_id}.wav'
self.output_audio_id += 1 self.output_audio_id += 1