Update README.md

README.md

@@ -12,6 +12,8 @@ The model of trained Qwen3 only processing chunk.
 import torch
 
 from transformers import AutoTokenizer, AutoModel
+from tqdm import tqdm
+from more_itertools import chunked
 
 residual = False
 residual_factor = 0.5
@@ -118,12 +120,17 @@ def encode_passage(tokenizer, model, pooling, passages, batch_size, normalize, m
         pas_embs_residual = None
     return pas_embs, pas_embs_residual
 
+your_query = "Your Query"
+
 query_hidden, _ = encode_query(
-    tokenizer, model, pooling_type="eos", queries=["Your query"],
+    tokenizer, model, pooling_type="eos", queries=[your_query],
     batch_size=8, normalize=True, max_length=8192, residual=residual,
 )
+
+your_chunk = "Your Chunk"
+
 candidate_hidden, candidate_hidden_residual = encode_passage(
-    tokenizer, model, pooling_type="eos", passages=["Your chunk<|endoftext|>Your context"],
+    tokenizer, model, pooling_type="eos", passages=[your_chunk],
     batch_size=4, normalize=True, max_length=8192, residual=residual,
 )