orionweller commited on
Commit
c27007f
·
1 Parent(s): a652c50

lower batch size

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 32
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]