Skip to content

Commit 257632e

Browse files
compatible with old versions of modelscope (#671)
1 parent e4f47bc commit 257632e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

swift/llm/utils/dataset.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,8 @@ def load_ms_dataset(
233233
subset_name, split = subset_split
234234
dataset = MsDataset.load(
235235
dataset_id, subset_name=subset_name, split=split)
236+
if hasattr(dataset, 'to_hf_dataset'):
237+
dataset = dataset.to_hf_dataset()
236238
dataset_list.append(dataset)
237239
return concatenate_datasets(dataset_list)
238240

0 commit comments

Comments
 (0)