fix import error

modelscope · May 15, 2024 · f29c475 · f29c475
1 parent 43cdafb
commit f29c475
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/modelscope/metrics/__init__.py b/modelscope/metrics/__init__.py
@@ -69,7 +69,9 @@
         'loss_metric': ['LossMetric'],
         'image_colorization_metric': ['ImageColorizationMetric'],
         'ocr_recognition_metric': ['OCRRecognitionMetric'],
-        'translation_evaluation_metric': ['TranslationEvaluationMetric']
+        'translation_evaluation_metric': ['TranslationEvaluationMetric'],
+        'video_super_resolution_metric.video_super_resolution_metric':
+        ['VideoSuperResolutionMetric'],
     }
 
     import sys

diff --git a/modelscope/trainers/__init__.py b/modelscope/trainers/__init__.py
@@ -36,7 +36,7 @@
         'nlp_trainer': ['NlpEpochBasedTrainer', 'VecoTrainer'],
         'trainer': ['EpochBasedTrainer'],
         'training_args': ['TrainingArgs', 'build_dataset_from_file'],
-        'hooks': ['Hook']
+        'hooks': ['Hook', 'Priority']
     }
 
     import sys