fix (modelscope#1232)

hjh0119 · Jun 26, 2024 · cce6bd7 · cce6bd7
1 parent 253d0ce
commit cce6bd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/llm/utils/dataset.py b/swift/llm/utils/dataset.py
@@ -1141,7 +1141,7 @@ def preprocess(row):
 
 
 register_dataset(
-    DatasetName.refcoco_unofficial,
+    DatasetName.refcoco_unofficial_caption,
     'swift/refcoco', [],
     preprocess_func=preprocess_refcoco_unofficial_caption,
     get_function=get_dataset_from_repo,