罗俊辉 преди 1 година
родител
ревизия
693d2122fe
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      alg_app.py

+ 1 - 1
alg_app.py

@@ -16,7 +16,7 @@ async def init():
     初始化模型
     """
     await AsyncMySQL.init_pool()
-    model = BertSimilarity(model_name_or_path="BAAI/bge-large-zh-v1.5")
+    model = BertSimilarity(model_name_or_path="BAAI/bge-large-zh-v1.5", device="cuda")
     print("模型加载成功")
     app_routes = AlgRoutes(AsyncMySQL, model)
     app.register_blueprint(app_routes)