[enh] Add support for large_string in model card create

UKPLab · Oct 18, 2024 · 9f61e38 · 9f61e38
1 parent 1802076
commit 9f61e38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sentence_transformers/model_card.py b/sentence_transformers/model_card.py
@@ -423,7 +423,7 @@ def set_widget_examples(self, dataset: Dataset | DatasetDict) -> None:
             columns = [
                 column
                 for column, feature in dataset[dataset_name].features.items()
-                if isinstance(feature, Value) and feature.dtype == "string" and column != "dataset_name"
+                if isinstance(feature, Value) and (feature.dtype == "string" or feature.dtype == "large_string") and column != "dataset_name"
             ]
             str_dataset = dataset[dataset_name].select_columns(columns)
             dataset_size = len(str_dataset)