This commit is contained in:
yumoqing 2025-06-19 13:41:18 +08:00
parent b6219805e6
commit 8f8502cb9e

View File

@ -52,6 +52,7 @@ def main():
raise e raise e
se = ServerEnv() se = ServerEnv()
se.chat_engine = Klass(args.model_path) se.chat_engine = Klass(args.model_path)
se.chat_engine.use_mps_if_prosible()
workdir = args.workdir or os.getcwd() workdir = args.workdir or os.getcwd()
port = args.port port = args.port
webserver(init, workdir, port) webserver(init, workdir, port)