Skip to content

Commit 5e1a7a4

Browse files
authored
[enh] Add support for large_string in model card create (#2999)
* [enh] Add support for large_string in model card create * [enh] Add support for large_string in model card create, with pre-commit checked
1 parent 2fa3ed4 commit 5e1a7a4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sentence_transformers/model_card.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,9 @@ def set_widget_examples(self, dataset: Dataset | DatasetDict) -> None:
423423
columns = [
424424
column
425425
for column, feature in dataset[dataset_name].features.items()
426-
if isinstance(feature, Value) and feature.dtype == "string" and column != "dataset_name"
426+
if isinstance(feature, Value)
427+
and (feature.dtype == "string" or feature.dtype == "large_string")
428+
and column != "dataset_name"
427429
]
428430
str_dataset = dataset[dataset_name].select_columns(columns)
429431
dataset_size = len(str_dataset)

0 commit comments

Comments
 (0)