fix

4905ab7a · zhangwq5 · faf0f8ff · 4905ab7a
Commit 4905ab7a authored Jul 16, 2025 by zhangwq5
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

infer/infer_vllm.py infer/infer_vllm.py +1 -1

No files found.
--- a/infer/infer_vllm.py
+++ b/infer/infer_vllm.py
@@ -52,7 +52,7 @@ def main(args: Namespace):
        print("-" * 60)

    script_dir = os.path.dirname(os.path.abspath(__file__))
-    output_path = os.path.join(script_dir, 'embeddings_A800.npy')
+    output_path = os.path.join(script_dir, 'embeddings_K100_AI.npy')
    np.save(output_path, np.array(embeddings))
    print(f"所有嵌入已保存到: {output_path}")