From fce4e4663511218ab0bfd8af47dbd735468cba5d Mon Sep 17 00:00:00 2001 From: hanhainebula <2512674094@qq.com> Date: Sat, 19 Oct 2024 20:28:04 +0800 Subject: [PATCH] del debug code for FlagLLMModel inference --- examples/inference/embedder/decoder_only/base_single_device.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/examples/inference/embedder/decoder_only/base_single_device.py b/examples/inference/embedder/decoder_only/base_single_device.py index a11f039..be1d007 100644 --- a/examples/inference/embedder/decoder_only/base_single_device.py +++ b/examples/inference/embedder/decoder_only/base_single_device.py @@ -25,9 +25,6 @@ def test_base_single_device(): queries_embeddings = model.encode_queries(queries) passages_embeddings = model.encode_corpus(passages) - print(queries_embeddings) - print(passages_embeddings) - cos_scores = queries_embeddings @ passages_embeddings.T print(cos_scores[:2, :2])