From 0c2fb4793c163e0d1b8a1eaeaed98c4b9ea79fce Mon Sep 17 00:00:00 2001 From: liaoxiju Date: Mon, 9 Dec 2024 08:51:09 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E5=AD=97=E4=BA=BA=E6=9C=8D=E5=8A=A1?= =?UTF-8?q?=E6=8F=90=E4=BA=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- auto_deploy_dight_human.sh | 29 ++++ common/__init__.py | 26 ++++ common/config.py | 218 ++++++++++++++++++++++++++++ common/security_check.py | 40 +++++ common/use_model.py | 87 +++++++++++ configs/cfg.yml | 10 ++ configs/config-template.j2 | 9 ++ configs/config-vars-dev.yml | 10 ++ configs/config-vars-mindie.yml | 10 ++ configs/config-vars-prod.yml | 10 ++ configs/config-vars-test.yml | 10 ++ digit_human.conf.py | 30 ++++ main.py | 76 ++++++++++ model_utils.py | 42 ++++++ requirements-sadtalker.txt | 0 requirements.txt | 0 sadtalker-server/start_sadtalker.sh | 0 sadtalker-server/stop_sadtalker.sh | 0 start.sh | 8 + stop.sh | 1 + test.py | 3 + 21 files changed, 619 insertions(+) create mode 100644 auto_deploy_dight_human.sh create mode 100644 common/__init__.py create mode 100644 common/config.py create mode 100644 common/security_check.py create mode 100644 common/use_model.py create mode 100644 configs/cfg.yml create mode 100644 configs/config-template.j2 create mode 100644 configs/config-vars-dev.yml create mode 100644 configs/config-vars-mindie.yml create mode 100644 configs/config-vars-prod.yml create mode 100644 configs/config-vars-test.yml create mode 100644 digit_human.conf.py create mode 100644 main.py create mode 100644 model_utils.py create mode 100644 requirements-sadtalker.txt create mode 100644 requirements.txt create mode 100644 sadtalker-server/start_sadtalker.sh create mode 100644 sadtalker-server/stop_sadtalker.sh create mode 100644 start.sh create mode 100644 stop.sh diff --git a/auto_deploy_dight_human.sh b/auto_deploy_dight_human.sh new file mode 100644 index 0000000..7e61b84 --- /dev/null +++ b/auto_deploy_dight_human.sh @@ -0,0 +1,29 @@ +#!/bin/bash +# 定义logs文件夹的名称 +LOG_DIR="logs" + +cd /data/redserver/red-agent-service/scene-digit-human +export HAIRUO_ENV=prod + +# 检查logs文件夹是否存在 +if [ ! -d "$LOG_DIR" ]; then + # 如果logs文件夹不存在，则创建它 + echo "Creating logs directory..." + mkdir "$LOG_DIR" +fi +# 设置conda环境名 +conda_env_name="agent-common" +echo "staring upload_api..." + +# 使用命令替换和if语句来检查conda环境中是否存在 +if conda env list | grep -q "$conda_env_name"; then + # 如果存在，则先杀掉进程 + ps -ef | grep upload_api | grep -v grep | awk '{print $2}'| xargs kill -9 2>/dev/null || true + python /data/config-manager/generate_service_configs.py --service_config_info_path configs/config-vars.yml --config_path configs/cfg.yml + conda run -n "$conda_env_name" gunicorn main:app -n digithuiman_api -c digithuman.conf.py --daemon + echo "Conda environment name is set to: $conda_env_name" + echo "Gunicorn started in the background in $conda_env_name environment." +else + # 如果不存在，则打印找不到 + echo "can not find : $conda_env_name" +fi \ No newline at end of file diff --git a/common/__init__.py b/common/__init__.py new file mode 100644 index 0000000..90100c3 --- /dev/null +++ b/common/__init__.py @@ -0,0 +1,26 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# +# Copyright @2024 INSPUR Inc. (inspur.com) +# +# @author: J.G. Chen +# @date: 2024/06/14 +# + +import os +from enum import Enum + +__version__ = "v1.4.2" + + +class HairuoEnv(str, Enum): + UNK = "unk" + TEST = "test" + DEV = "dev" + PROD = "prod" + + +# 默认所有的机器需要设置 "HAIRUO_ENV" 环境变量, test/dev/prod +HAIRUO_ENV = HairuoEnv(os.getenv("HAIRUO_ENV", "unk")) + +assert HAIRUO_ENV != HairuoEnv.UNK, "env var `HAIRUO_ENV` is required." diff --git a/common/config.py b/common/config.py new file mode 100644 index 0000000..431433d --- /dev/null +++ b/common/config.py @@ -0,0 +1,218 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# +# Copyright @2024 INSPUR Inc. (inspur.com) +# +# @author: J.G. Chen +# @date: 2024/02/17 +# +""" +对配置文件的一些操作 + +Notes + - pyyaml: + * 对科学计数法的支持有特殊要求，对于 a[eE][+-]b：a 必须有小数点，指数必须含正负号 + `pyyaml issues#173 `_; + * 或者使用 ruamel.yaml 代替； + - conf(hocon): + * 变量替换：不能添加引号，`{ b = "hello"\n a = ${b} world }`; + * 环境变量：'{ a = ${HOME} }'; +""" + +import json +import os +from pathlib import Path +from platform import python_version +from pprint import pformat +from typing import Dict +from typing import Union + +import yaml +from omegaconf import DictConfig +from omegaconf import ListConfig +from omegaconf import OmegaConf +from packaging import version +from pyhocon import ConfigFactory +from pyhocon import HOCONConverter + +from common import HAIRUO_ENV +from common import HairuoEnv + +try: + import torch +except ImportError: + torch = None + +try: + from common.utils.logger import init_logger + + logger = init_logger(__name__).info +except ImportError: + init_logger = None + logger = print + + +class ConfigBase: + config: Union[DictConfig, ListConfig] + + def __init__(self, config_file: str = ""): + """load config from `config_file` + + Args: + config_file(str): config_file name, could be `.yaml`, `.conf`, `.json`, `.bin` + Notes: + config_file is relative to current work dir(cwd), pass to abs path + """ + self._file_path = self._resolve(config_file) + self.config = self.load(self._file_path) + + @classmethod + def _resolve(cls, config_file: str = ""): + cwd = Path.cwd() + valid_extension = (".yaml", ".yml", ".conf", ".json", ".bin") + + # check config file in current work dir + if not config_file: + files = filter(lambda f: f.is_file(), cwd.iterdir()) + logger(f"loading config file from dir: {cwd.resolve()}") + for filename in sorted(files, key=lambda x: os.path.getmtime(x), reverse=True): + if filename.suffix in valid_extension: + config_file = Path(cwd, filename) + break + else: + # parse config file specified. + config_file = Path(config_file) + + # check config file + if not (config_file and config_file.is_file() and config_file.suffix in valid_extension): + raise FileNotFoundError(f"Config '{config_file}' not find!") + + logger(f"loading config from {config_file.resolve()}") + return config_file + + @classmethod + def show(cls, config: Union[Dict, DictConfig, ListConfig] = None): + """show config given or parsed self.config.""" + kwargs = {} if version.parse(python_version()) < version.parse("3.8") else {"sort_dicts": False} + if not config: + config = cls.config + + if isinstance(config, (DictConfig, ListConfig)): + config = OmegaConf.to_object(config) + + logger(f"======= resolved config ======>\n{pformat(config, **kwargs).encode('utf-8')}") + + @classmethod + def load(cls, path: Union[str, Path]) -> Union[DictConfig, ListConfig]: + """load config file.""" + path = Path(path) + logger(f"loading config from: {path}") + + if path.suffix in [".yaml", ".yml"]: + with path.open(encoding="utf-8") as f: + config = OmegaConf.create(yaml.load(f, Loader=yaml.FullLoader), flags={"allow_objects": True}) + elif path.suffix == ".conf": + config = ConfigFactory.parse_file(path.as_posix()) + config = OmegaConf.create(HOCONConverter.to_json(config)) + elif path.suffix == ".json": + with path.open() as f: + config = OmegaConf.create(json.load(f)) + elif path.suffix == ".bin": + if torch is not None: + config = OmegaConf.create(torch.load(path)) + else: + raise RuntimeError("`torch` required to load .bin config file.") + else: + raise RuntimeError("unsupported file format to load.") + + logger(f"parse configs for ENV: {HAIRUO_ENV}") + # keep configs for given env only + # TODO: update fields recursively + config.update(config.get(HAIRUO_ENV, dict())) + for env in HairuoEnv: + config.pop(env, default=None) + + ConfigBase.show(config) + return config + + @classmethod + def save_config(cls, config: Union[dict, DictConfig, ListConfig], path: Union[str, Path]): + """save `config` to given `path`""" + path = Path(path) + + def convert_to_ct(_config): + # convert dict to ConfigTree recursively. + if isinstance(_config, (dict, DictConfig)): + tmp = {} + for k, v in _config.items(): + tmp[k] = convert_to_ct(v) + return ConfigFactory.from_dict(tmp) + elif isinstance(_config, ListConfig): + return [convert_to_ct(i) for i in _config] + else: + return _config + + if path.suffix in [".yaml", ".yml"]: + OmegaConf.save(config, path) + elif path.suffix == ".conf": + with open(path, "w") as writer: + writer.write(HOCONConverter.to_hocon(convert_to_ct(config))) + writer.write("\n") + elif path.suffix == ".json": + with open(path, "w") as writer: + json.dump(config, writer, ensure_ascii=False, indent=2) + elif path.suffix == ".bin": + if torch is not None: + torch.save(config, path) + else: + raise RuntimeError("`torch` required to load .bin config file.") + else: + raise RuntimeError("unsupported file format.") + logger(f"saving config to: {path}") + + def save(self, path: Union[str, Path] = "", key: str = ""): + """save resolved config obj to `path` or path provided by `key` in config file. + + Args: + path: save + key: the `path` is resolved from `key` in resolved config. + """ + + if path: + pass + elif key and self.config.get(key, ""): + path = self.config.get(key) + else: + raise RuntimeError("target dir not found") + + if Path(path).is_dir(): + path = Path(path) / self._file_path.name + else: + path = Path(path) + + self.save_config(self.config, path) + + +if __name__ == "__main__": + import argparse + + parser = argparse.ArgumentParser( + description="tool to convert config file format", + formatter_class=argparse.ArgumentDefaultsHelpFormatter, + ) + args = parser.add_argument + args("path", metavar="config_file", help="path to config file.") + args("-f", "--format", type=str, default="conf", choices=["conf", "yaml", "json"], help="save conf in format.") + args("-o", "--output", type=str, help="output config to file.") + args("-s", "--show", action="store_true", help="just show the resolved config.") + params = parser.parse_args() + + conf = ConfigBase.load(params.path) + output = params.output + if not output: + output = Path(params.path).with_suffix(f".{params.format.strip('.')}") + + if params.show: + ConfigBase.show(conf) + else: + ConfigBase.save_config(conf, output) diff --git a/common/security_check.py b/common/security_check.py new file mode 100644 index 0000000..05e05f3 --- /dev/null +++ b/common/security_check.py @@ -0,0 +1,40 @@ +import requests + + +def security_check(SECURITY_URL: str, auth_code: str, question: str, isRejection: bool, isRefusal: bool): + ''' + SECURITY_URL: 拒识拒答url + auth_code: token + question: 被检测的字符串 + isRejection: 是否进行拒识检测 + isRefusal: 是否进行拒答检测 + threshold: 知识库检索最低阈值 + ''' + headers = { + 'content-type': "application/json", + 'authorization': auth_code, + } + security_json = { + "query": question, + "isRejection": isRejection, + "isRefusal": isRefusal, + } + try: + security_res = requests.post(SECURITY_URL, json=security_json, headers=headers) + # {'code': 2, 'message': '拒识检测未通过!', 'result': False} + security_res_json = security_res.json() + if not security_res_json['result']: + return { + "result": False, + "msg": security_res_json["message"] + } + elif security_res_json['result']: + return { + "result": True, + "msg": security_res_json["message"] + } + except Exception as e: + return { + "result": False, + "msg": f"error: {e}" + } diff --git a/common/use_model.py b/common/use_model.py new file mode 100644 index 0000000..cda58fd --- /dev/null +++ b/common/use_model.py @@ -0,0 +1,87 @@ +import sys + +sys.path.append("..") +from utils.utils import get_logger +import time +import requests +import json + +logger = get_logger("hairuo_general_vllm") + + +def ask_question(query_url, messages, stream): + if stream: + return vllm_chat_stream(query_url, messages) + else: + return vllm_chat_non_flow(query_url, messages) + + +def vllm_chat_non_flow(query_url, messages): + try: + T1 = time.time() + inter_q = [ + { + "role": "system", + "content": "You are a helpful assistant." + } + ] + messages + post_json = { + "model": "general_model", + "stream": False, + "stop": "<|im_end|>", + "messages": inter_q + } + headers = { + 'Content-Type': 'application/json' + } + response = requests.post(query_url, headers=headers, json=post_json, timeout=60) + T2 = time.time() + logger.info('程序运行时间:%s毫秒' % ((T2 - T1) * 1000)) + logger.info(f'Ask result:{response.json()}') + if response.status_code == 200: + data_ok = response.json() + first_choice = data_ok['choices'][0] + message_content = first_choice['message']['content'] + return message_content + else: + return response.json() + except Exception as e: + logger.info(f"Ask occur an error: {e}") + return "Model running abnormally!" + + +def vllm_chat_stream(query_url, messages): + inter_q = [ + { + "role": "system", + "content": "You are a helpful assistant." + } + ] + messages + req_body = { + "model": 'general_model', + "stream": True, + "stop": "<|im_end|>", + "messages": inter_q + } + + headers = {'Content-Type': 'application/json;charset=utf-8'} + response = requests.post(query_url, json=req_body, headers=headers, stream=True) + answer = "" + # answer_result = [] + for chunk in response.iter_lines(): + if chunk: + chunk = chunk.decode('utf-8') + chunk = chunk.replace("data: ", "") + try: + try: + chunk = json.loads(chunk) + except: + logger.info("-----最后一条------------------------------------") + if type(chunk) != str: + if chunk['choices'][0]['delta'].get('content', '') != '': + answer = answer + chunk['choices'][0]['delta']['content'] + if chunk['choices'][0]["finish_reason"] != "stop": + yield answer + except Exception as e: + logger.info("流式输出error: %s", e) + return answer diff --git a/configs/cfg.yml b/configs/cfg.yml new file mode 100644 index 0000000..8506602 --- /dev/null +++ b/configs/cfg.yml @@ -0,0 +1,10 @@ +# 数字人 +dighthuman: + dev: + dh_webui: http://100.200.128.72:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + test: + dh_webui: http://100.200.128.72:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + prod: + dh_webui: http://100.200.128.72:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + model_name: HaiRuo-AudioVisual-ST + static_token: 7c3eafb5-2d6e-100d-ab0f-7b2c1cdafb3c diff --git a/configs/config-template.j2 b/configs/config-template.j2 new file mode 100644 index 0000000..9a6e140 --- /dev/null +++ b/configs/config-template.j2 @@ -0,0 +1,9 @@ +dighthuman: + dev: + dh_webui: {{ red_server.dighthuman.dev.dh_webui }} + test: + dh_webui: {{ red_server.dighthuman.test.dh_webui }} + prod: + dh_webui: {{ red_server.dighthuman.prod.dh_webui }} + model_name: {{ red_server.dighthuman.model_name }} + static_token: {{ global.authorization.static_token }} \ No newline at end of file diff --git a/configs/config-vars-dev.yml b/configs/config-vars-dev.yml new file mode 100644 index 0000000..6080a9a --- /dev/null +++ b/configs/config-vars-dev.yml @@ -0,0 +1,10 @@ +red_server: + dighthuman: + dev: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + test: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + prod: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + model_name: HaiRuo-AudioVisual-ST + static_token: 7c3eafb5-2d6e-100d-ab0f-7b2c1cdafb3c \ No newline at end of file diff --git a/configs/config-vars-mindie.yml b/configs/config-vars-mindie.yml new file mode 100644 index 0000000..6080a9a --- /dev/null +++ b/configs/config-vars-mindie.yml @@ -0,0 +1,10 @@ +red_server: + dighthuman: + dev: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + test: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + prod: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + model_name: HaiRuo-AudioVisual-ST + static_token: 7c3eafb5-2d6e-100d-ab0f-7b2c1cdafb3c \ No newline at end of file diff --git a/configs/config-vars-prod.yml b/configs/config-vars-prod.yml new file mode 100644 index 0000000..6080a9a --- /dev/null +++ b/configs/config-vars-prod.yml @@ -0,0 +1,10 @@ +red_server: + dighthuman: + dev: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + test: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + prod: + dh_webui: http://127.0.0.1:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + model_name: HaiRuo-AudioVisual-ST + static_token: 7c3eafb5-2d6e-100d-ab0f-7b2c1cdafb3c \ No newline at end of file diff --git a/configs/config-vars-test.yml b/configs/config-vars-test.yml new file mode 100644 index 0000000..5352bf4 --- /dev/null +++ b/configs/config-vars-test.yml @@ -0,0 +1,10 @@ +red_server: + dighthuman: + dev: + dh_webui: http://100.200.128.83:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + test: + dh_webui: http://100.200.128.83:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + prod: + dh_webui: http://100.200.128.83:14040/agentstore/api/v1/multimodal_models/dh/dighthuman + model_name: HaiRuo-AudioVisual-ST + static_token: 7c3eafb5-2d6e-100d-ab0f-7b2c1cdafb3c \ No newline at end of file diff --git a/digit_human.conf.py b/digit_human.conf.py new file mode 100644 index 0000000..8b7b538 --- /dev/null +++ b/digit_human.conf.py @@ -0,0 +1,30 @@ + # -*- coding: utf-8 -*- + +import os + +path_of_current_file = os.path.abspath(__file__) +path_of_current_dir = os.path.split(path_of_current_file)[0] + +# worker_class为sync会报错 +# uvicorn.workers.UvicornWorker +worker_class = 'uvicorn.workers.UvicornWorker' +# workers = multiprocessing.cpu_count() * 2 + 1 +workers = 1 # 按需启动的进程数 +threads = 1 # 各进程包含的线程数 + +chdir = path_of_current_dir + +worker_connections = 0 +timeout = 0 +max_requests = 0 +graceful_timeout = 0 + +loglevel = 'info' +access_log_format = '%(t)s %(p)s %(h)s "%(r)s" %(s)s %(L)s %(b)s %(f)s" "%(a)s"' +reload = True +debug = False +bind = "%s:%s" % ("0.0.0.0", 14040) +pidfile = '%s/digithuiman.pid' % (path_of_current_dir) +errorlog = '%s/logs/digithuiman.log' % (path_of_current_dir) +accesslog = '%s/logs/digithuiman_access.log' % (path_of_current_dir) +proc_name = "digithuiman_api" diff --git a/main.py b/main.py new file mode 100644 index 0000000..d5b58cb --- /dev/null +++ b/main.py @@ -0,0 +1,76 @@ +# -*- encoding: utf-8 -*- +''' +@Email : liaoxiju@inspur.com +''' +import os +import re +import sys +sys.path.append("../") +sys.path.append("../..") +import datetime +from datetime import timedelta +import logging +import yaml +import uvicorn +from common import HAIRUO_ENV +from common import HairuoEnv +from fastapi import FastAPI, Request +from agent_common_utils.logger import get_logger +from agent_common_utils.function_monitor import log_function_call +from model_utils import audio_driven_video + +logger = get_logger("digithuman") + +dh_audio_driven_video = log_function_call(logger)(audio_driven_video) + +app = FastAPI() + + +@app.post("/hairuo/digithuman") +async def digithuman(request: Request, req: dict): + ''' + 对输入文本文本描述，生成wav音频，然后调用dh-webui生成视频 + ''' + + logger.info(" Verification authorization.") + body_data = await request.body() + body_data = body_data.decode("utf-8") + + logger.info(" {}".format(body_data)) + logger.info(" input text.") + try: + app_id = req.get("app_id") + prompt = req.get("text") + + ##edge tts生成音频，上传oss，获取音频url。 TODO:待实现 + wav_url = 'xxx' + + video_url, code = dh_audio_driven_video(wav_url) + if not code: + logger.info(" dh model error") + resp = { + "code": "1", + "message": "dh model error", + "result": '' + } + return resp + resp = { + "app_id": app_id, + "code": "0", + "message": "success", + "result": video_url + } + return resp + + except: + logger.info(" model error") + resp = { + "code": "1", + "message": "dh model call error", + "result": "" + } + return resp + + +if __name__ == "__main__": + uvicorn.run(app, host="0.0.0.0", port=14040) \ No newline at end of file diff --git a/model_utils.py b/model_utils.py new file mode 100644 index 0000000..fb4545b --- /dev/null +++ b/model_utils.py @@ -0,0 +1,42 @@ +# -*- encoding: utf-8 -*- +''' +@File : model_utils.py +@Time : 2024/07/24 10:28:28 +@Author : liangzz1991 +@Email : zhaoliang03@inspur.com +''' +import json +import time +import yaml +import requests +from common.config import ConfigBase +from common import HAIRUO_ENV +from common import HairuoEnv +from agent_common_utils.logger import get_logger + +logger = get_logger("hairuo_text2img_sd-webui") +def get_cfg(config): + config = config.text2img + config.update(config.get(HAIRUO_ENV, dict())) + for env in HairuoEnv: + config.pop(env, default=None) + ConfigBase.show(config) + return config + +configs = ConfigBase.load('configs/cfg.yml') +configs = get_cfg(configs) +STATIC_TOKEN = configs['static_token'] +DH_URL = configs['hu_webui'] +model_name = configs['model_name'] + +def audio_driven_video(wav_url): + data = { + "audio_url": wav_url + } + response = requests.post(DH_URL, data=json.dumps(data),timeout=60,headers={'Content-Type': 'application/json', 'Authorization': STATIC_TOKEN}) + if response.status_code != 200: + return 1, "" + else: + video_url = response.json()['result'] + return 0, video_url + diff --git a/requirements-sadtalker.txt b/requirements-sadtalker.txt new file mode 100644 index 0000000..e69de29 diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..e69de29 diff --git a/sadtalker-server/start_sadtalker.sh b/sadtalker-server/start_sadtalker.sh new file mode 100644 index 0000000..e69de29 diff --git a/sadtalker-server/stop_sadtalker.sh b/sadtalker-server/stop_sadtalker.sh new file mode 100644 index 0000000..e69de29 diff --git a/start.sh b/start.sh new file mode 100644 index 0000000..76c06fe --- /dev/null +++ b/start.sh @@ -0,0 +1,8 @@ +SELF_DIR=$(cd $(dirname "$0"); pwd) +PROJECT_ROOT=${SELF_DIR}/.. +export PYTHONPATH=$PROJECT_ROOT:$PYTHONPATH +cd $PROJECT_ROOT/scene-digit-human +# python main.py +mkdir logs +python /data/config-manager/generate_service_configs.py --service_config_info_path configs/config-vars.yml --config_path configs/cfg.yml +gunicorn main:app -n digithuman_api -c digit_human.conf.py --daemon \ No newline at end of file diff --git a/stop.sh b/stop.sh new file mode 100644 index 0000000..6fa8870 --- /dev/null +++ b/stop.sh @@ -0,0 +1 @@ +ps -ef | grep digithuman_api | grep -v grep | awk '{print $2}'| xargs kill -9 \ No newline at end of file diff --git a/test.py b/test.py index f33e7c2..246d091 100644 --- a/test.py +++ b/test.py @@ -1,3 +1,6 @@ +""" +@Email: liaoxiju@inspur.com +""" from modelscope.piplines import pipeline # Create a pipeline instance for talking head generation using the specified model and revision.