File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed
src/optimum/rbln/transformers/models/gemma3 Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments