Spaces:

HF-test-lab
/

bulk_embeddings

Runtime error

nbroad commited on Jul 14, 2023

Commit

41912a4

1 Parent(s): d02045d

don't remove columns

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -284,7 +284,6 @@ def batch_embed(
                 "column_name": column_name,
                 "padding": "max_length" if opt_level == "O4" else True,
             },
-            remove_columns=ds.column_names,
         )
     )

                 "column_name": column_name,
                 "padding": "max_length" if opt_level == "O4" else True,
             },
         )
     )