Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

gemma3 in grpo doesn't work when using lmdeploy #3785

Closed
kakao-charlie-cs opened this issue Apr 7, 2025 · 2 comments
Closed

gemma3 in grpo doesn't work when using lmdeploy #3785

kakao-charlie-cs opened this issue Apr 7, 2025 · 2 comments

Comments

@kakao-charlie-cs
Copy link

kakao-charlie-cs commented Apr 7, 2025

Describe the bug
What the bug is, and how to reproduce, better with screenshots(描述bug以及复现过程,最好有截图)

I followed this example script for training gemma3-4b in GRPO

I used gemma3-4b, but no-vision model. So it is generally similar as gemma-3-1b-it.

lmdeploy model which is not supported by turbomind doesn't have load_weights method.
Therefore, below lines raised Exception related the method.

if self.infer_rank >= 0:
if self.args.async_generate:
self._wait_queue()
if self.args.use_vllm:
llm_model = self.engine.inner_model
else:
llm_model = self.engine.engine.engine
llm_model.load_weights(state_dict.items())

qwen2_5 based model works fine (since it was supported). But gemma3 not work properly.

Your hardware and system info
Write your system info like CUDA version/system/GPU/torch version here(在这里给出硬件信息和系统信息,如CUDA版本,系统,GPU型号和torch版本等)

  • single node H200-8ea
  • deepspeed==0.14.5
  • trl==0.16.1
  • lmdeploy==0.7.2.post1
  • torch==2.6.0
  • CUDA==12.2
  • ms-swift==3.3.0dev

Additional context
Add any other context about the problem here(在这里补充其他信息)

@RomanticGodVAN
Copy link

the same issue

@hjh0119
Copy link
Collaborator

hjh0119 commented Apr 8, 2025

The integration with LMDeploy currently only works with the turbomind backend, for non-TurboMind compatible models: plz use vLLM or pt backend

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants