fix

hiworldwzj · hiworldwzj · commit 7c664c39bc1a · 2026-05-20T07:49:34.000Z
diff --git a/lightllm/models/gemma4/model.py b/lightllm/models/gemma4/model.py
@@ -155,6 +155,9 @@ def _init_ple_static_buffer(self):
         )
 
     def _init_to_get_rotary_gemma4(self):
+        # gemma4 当前不支持 dp prefill balance
+        assert self.args.enable_dp_prefill_balance is False, "Gemma-4 does not support dp prefill balance"
+
         rope_params = self.config["rope_parameters"]
 
         # Cap the rotary table at something we can fit in memory — Gemma-4's

Original file line number	Diff line number	Diff line change
`@@ -155,6 +155,9 @@ def _init_ple_static_buffer(self):`
`155`	`155`	`)`
`156`	`156`
`157`	`157`	`def _init_to_get_rotary_gemma4(self):`
	`158`	`+ # gemma4 当前不支持 dp prefill balance`
	`159`	`+ assert self.args.enable_dp_prefill_balance is False, "Gemma-4 does not support dp prefill balance"`
	`160`	`+`
`158`	`161`	`rope_params = self.config["rope_parameters"]`
`159`	`162`
`160`	`163`	`# Cap the rotary table at something we can fit in memory — Gemma-4's`