[Fix] Resolve recursive imports for metric recorder

Author: nil0x9

xtuner/v1/train/trainer.py

@@ -41,14 +41,13 @@
 from xtuner.v1.profiler.prober_utils import setup_prober_list
 from xtuner.v1.utils import (
     XTUNER_DETERMINISTIC,
-    InternalMetrics,
-    InternalMetricsRecorder,
     ParallelConfigException,
     get_logger,
     is_hf_model_path,
     log_format,
     record_git_info,
 )
+from xtuner.v1.utils.internal_metrics import InternalMetrics, InternalMetricsRecorder
 from xtuner.v1.utils.check_health import check_health
 from xtuner.v1.utils.device import get_device, get_torch_device_module
 

xtuner/v1/utils/__init__.py

@@ -5,7 +5,6 @@
 from .enum_helper import StrEnum
 from .exception_helper import ParallelConfigException
 from .init_weight import default_init_weights, init_params
-from .internal_metrics import InternalMetrics, InternalMetricsRecorder
 from .loader import HFCheckpointLoader
 from .logger import get_logger, log_format
 from .misc import XTUNER_DETERMINISTIC, SharedMemory, get_padding_length, is_hf_model_path, record_git_info
@@ -46,6 +45,4 @@
     "default_init_weights",
     "IGNORE_INDEX",
     "monkey_unpatch_torch_reductions",
-    "InternalMetricsRecorder",
-    "InternalMetrics",
 ]