|
@@ -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)
|