Spaces:
Sleeping
Sleeping
Commit
·
c27007f
1
Parent(s):
a652c50
lower batch size
Browse files
app.py
CHANGED
@@ -132,7 +132,7 @@ def encode_queries(dataset_name, postfix):
|
|
132 |
input_texts = [f"query: {query.strip()} {postfix}".strip() for query in queries[dataset_name]]
|
133 |
|
134 |
encoded_embeds = []
|
135 |
-
batch_size =
|
136 |
|
137 |
for start_idx in tqdm.tqdm(range(0, len(input_texts), batch_size), desc="Encoding queries"):
|
138 |
batch_input_texts = input_texts[start_idx: start_idx + batch_size]
|
|
|
132 |
input_texts = [f"query: {query.strip()} {postfix}".strip() for query in queries[dataset_name]]
|
133 |
|
134 |
encoded_embeds = []
|
135 |
+
batch_size = 2
|
136 |
|
137 |
for start_idx in tqdm.tqdm(range(0, len(input_texts), batch_size), desc="Encoding queries"):
|
138 |
batch_input_texts = input_texts[start_idx: start_idx + batch_size]
|