Skip to content

Commit

Permalink
Update api.py
Browse files Browse the repository at this point in the history
  • Loading branch information
zhangkai727 authored Oct 28, 2024
1 parent 4ed3f31 commit b8e8257
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions img/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,12 +349,12 @@ def parse_args():
parser.add_argument(
"--llama-checkpoint-path",
type=str,
default="/home/sunzhu/Real-time-hyperpersonification/fish_audio_v2/checkpoints/fish-speech-1.4",
default="XXXX/fish_audio_v2/checkpoints/fish-speech-1.4", ## 模型加载地址
)
parser.add_argument(
"--decoder-checkpoint-path",
type=str,
default="/home/sunzhu/Real-time-hyperpersonification/fish_audio_v2/checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
default="/XXX/fish_audio_v2/checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth", ## 加载模型文件地址
)
parser.add_argument("--decoder-config-name", type=str, default="firefly_gan_vq")
parser.add_argument("--device", type=str, default="cuda")
Expand Down Expand Up @@ -439,7 +439,7 @@ async def data(self) -> Annotated[Any, ContentType("application/msgpack")]:
)
)

url = '10.27.127.33:8085'
url = 'XXX.XXX.XXX:8085' ## 设定启动API的地址和端口
logger.info(f"Warming up done, starting server at http://{url}")
host, port = url.split(":")
uvicorn.run(app, host=host, port=int(port), workers=args.workers, log_level="info")
Expand All @@ -458,4 +458,4 @@ async def data(self) -> Annotated[Any, ContentType("application/msgpack")]:
'''
# ps -ef | grep api | grep -v grep | awk '{print "kill "$2}' | sh
# tmux new-session -s fish
# sudo kill 75253
# sudo kill 75253

0 comments on commit b8e8257

Please sign in to comment.