fix tp lmdeploy (#1654)

Author: Jintao-Huang

swift/llm/utils/argument.py

@@ -507,8 +507,11 @@ def load_from_ckpt_dir(self, is_sft: bool = False) -> None:
         imported_keys = [
             'model_type', 'model_revision', 'template_type', 'dtype', 'quant_method', 'quantization_bit',
             'bnb_4bit_comp_dtype', 'bnb_4bit_quant_type', 'bnb_4bit_use_double_quant', 'model_id_or_path',
-            'custom_register_path', 'custom_dataset_info', 'tp', 'pp'
+            'custom_register_path', 'custom_dataset_info'
         ]
+        if (isinstance(self, SftArguments) and self.train_backend == 'megatron'
+                or isinstance(self, ExportArguments) and self.to_hf is True):
+            imported_keys += ['tp', 'pp']
         if not is_sft:
             imported_keys += ['sft_type', 'rope_scaling', 'system']
             if getattr(self, 'load_dataset_config', False) and from_sft_args:

swift/llm/utils/model.py

@@ -4309,6 +4309,7 @@ def new_get_rank(group=None):
     eos_token='<|im_end|>',
     support_flash_attn=True,
     support_lmdeploy=True,
+    requires=['decord'],
     tags=['multi-modal', 'vision'],
     function_kwargs={'version': 'v2.5'},
     hf_model_id='internlm/internlm-xcomposer2d5-7b')