Spaces:

HF-test-lab
/

bulk_embeddings

Runtime error

nbroad commited on Jul 15, 2023

Commit

43b32e0

1 Parent(s): b93ed22

text column name

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -340,7 +340,7 @@ def batch_embed(
         outputs = model(input_ids=ids, attention_mask=mask, token_type_ids=t_ids)
         embeds.extend(mean_pooling(outputs[0], mask).cpu().tolist())
-        texts.extend([b[column_name] for b in batch])
         current_count += ids.shape[0]

         outputs = model(input_ids=ids, attention_mask=mask, token_type_ids=t_ids)
         embeds.extend(mean_pooling(outputs[0], mask).cpu().tolist())
+        texts.extend(batch[column_name])
         current_count += ids.shape[0]