Fix lmdeploy branch (#3145)

modelscope · Feb 19, 2025 · 7989581 · 7989581
1 parent d7ec5a2
commit 7989581
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/examples/train/grpo/full_lmdeploy.sh b/examples/train/grpo/full_lmdeploy.sh
@@ -1,8 +1,8 @@
 # pip install lmdeploy==0.6.4
 # Replace three files:
-# 1. https://github.com/tastelikefeet/lmdeploy/blob/feat/reload_state_dict/lmdeploy/messages.py
-# 2. https://github.com/tastelikefeet/lmdeploy/blob/feat/reload_state_dict/lmdeploy/turbomind/turbomind.py
-# 3. https://github.com/tastelikefeet/lmdeploy/blob/feat/reload_state_dict/lmdeploy/turbomind/deploy/loader.py
+# 1. https://github.com/tastelikefeet/lmdeploy/blob/feat/reload_state_dict_064/lmdeploy/messages.py
+# 2. https://github.com/tastelikefeet/lmdeploy/blob/feat/reload_state_dict_064/lmdeploy/turbomind/turbomind.py
+# 3. https://github.com/tastelikefeet/lmdeploy/blob/feat/reload_state_dict_064/lmdeploy/turbomind/deploy/loader.py
 
 CUDA_VISIBLE_DEVICES=0,1,2,3 \
 NPROC_PER_NODE=3 \

diff --git a/swift/trainers/rlhf_trainer/grpo_trainer.py b/swift/trainers/rlhf_trainer/grpo_trainer.py
@@ -154,17 +154,17 @@ def __init__(self,
                             max_model_len=args.vllm_max_model_len)
                     self.engine.default_template = self.template
                 elif use_lmdeploy:
-                    # https://github.com/tastelikefeet/lmdeploy.git@feat/reload_state_dict
+                    # https://github.com/tastelikefeet/lmdeploy.git@feat/reload_state_dict_064
                     # Compile:https://github.com/tastelikefeet/lmdeploy/blob/main/docs/en/get_started/installation.md
                     if not is_lmdeploy_available():
                         raise ImportError('Please install `pip install lmdeploy==0.6.4`'
                                           ' and replace three files with:\n'
                                           '1. https://github.com/tastelikefeet/lmdeploy/blob/feat/'
-                                          'reload_state_dict/lmdeploy/messages.py\n'
+                                          'reload_state_dict_064/lmdeploy/messages.py\n'
                                           '2. https://github.com/tastelikefeet/lmdeploy/blob/feat/'
-                                          'reload_state_dict/lmdeploy/turbomind/turbomind.py\n'
+                                          'reload_state_dict_064/lmdeploy/turbomind/turbomind.py\n'
                                           '3. https://github.com/tastelikefeet/lmdeploy/blob/feat/'
-                                          'reload_state_dict/lmdeploy/turbomind/deploy/loader.py\n')
+                                          'reload_state_dict_064/lmdeploy/turbomind/deploy/loader.py\n')
                     from swift.llm import LmdeployEngine
                     from swift.tuners import Swift
                     with Swift.grpo_context(model, self.template.processor):