Spaces:

HF-test-lab
/

bulk_embeddings

Runtime error

nbroad commited on Jul 22, 2023

Commit

ae4cd1b

1 Parent(s): f245c03

Update infer.py

Files changed (1) hide show

infer.py CHANGED Viewed

@@ -151,8 +151,6 @@ def collate_fn(examples, column_name, tokenizer):
     feature_cols = ["input_ids", "attention_mask"]
     features = [{k: x[k] for k in feature_cols} for x in examples]
-    print(features)
     tokenized = tokenizer.pad(
         features,
         padding=True,

     feature_cols = ["input_ids", "attention_mask"]
     features = [{k: x[k] for k in feature_cols} for x in examples]
     tokenized = tokenizer.pad(
         features,
         padding=True,