from embedl.models.gemma.modeling_flash_head import FlashHeadGemma3ForCausalLM