fixed the issue of repeatedly importing the mindspeed patch

Author: ji-huazhong

verl/workers/engine/mindspeed/transformer_impl.py

@@ -15,7 +15,10 @@
 import logging
 import os
 
-from mindspeed.megatron_adaptor import repatch
+try:
+    from mindspeed.megatron_adaptor import repatch
+except ImportError:
+    repatch = None
 
 from verl.trainer.config import CheckpointConfig
 from verl.workers.config import HFModelConfig, McoreEngineConfig, McoreOptimizerConfig

verl/workers/megatron_workers.py

@@ -28,7 +28,7 @@
 from omegaconf import DictConfig, OmegaConf
 
 try:
-    from mindspeed.megatron_adaptor import repatch
+    from verl.workers.engine.mindspeed.transformer_impl import repatch
 except ImportError:
     repatch = None