Skip to content

Commit cb43894

Browse files
authored
model: fix gemma3 with transformers 4.53.1 (#317)
1 parent 3b9111e commit cb43894

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/optimum/rbln/transformers/models/gemma3/modeling_gemma3.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -408,6 +408,13 @@ def _create_embedding_layer(self):
408408
def _update_sliding_window_config(cls, model_config: PretrainedConfig, rbln_config: RBLNGemma3ForCausalLMConfig):
409409
sliding_window = getattr(model_config, "sliding_window", None)
410410
sliding_window_pattern = getattr(model_config, "sliding_window_pattern", None)
411+
if sliding_window_pattern is None:
412+
if hasattr(model_config, "layer_types"):
413+
first_full_attention_index = model_config.layer_types.index("full_attention")
414+
sliding_window_pattern = first_full_attention_index + 1
415+
else:
416+
raise ValueError("Cannot determine sliding_window_pattern from model_config")
417+
411418
if sliding_window_pattern <= model_config.num_hidden_layers:
412419
rbln_config.cache_impl = "hybrid"
413420
rbln_config.sliding_window = sliding_window

0 commit comments

Comments
 (0)