增加一些设置。

2025-10-17 18:24:04 +08:00
parent 25f8e0a267
commit 5db96600fe
10 changed files with 38 additions and 25 deletions
@@ -197,8 +197,8 @@ def build_and_export_similarity_head(

 def main():
    print("开始转换检索头为移动端格式（TorchScript）...")
-    embeddings_path = "../faiss_vector_db/faiss_index/embeddings.json"
-    output_path = "../faiss_vector_db/faiss_index/similarity_head.pt"
+    embeddings_path = "../faiss_vector_db/DishClassification/faiss_index/embeddings.json"
+    output_path = "../faiss_vector_db/DishClassification/faiss_index/similarity_head.pt"

    try:
        path, n_items = build_and_export_similarity_head(embeddings_path, output_path)