Fix typing hints for different rope parameters per layer type (#43320)
* Fix typing hints for different rope parameters per layer type
Signed-off-by: Tcc0403 <76503978+Tcc0403@users.noreply.github.com>
* Stricter assumption
Signed-off-by: Tcc0403 <76503978+Tcc0403@users.noreply.github.com>
* Update modular files and check if keys are present
Signed-off-by: Tcc0403 <76503978+Tcc0403@users.noreply.github.com>
* make fix-repo
Signed-off-by: Tcc0403 <76503978+Tcc0403@users.noreply.github.com>
* fix repo
* run modular gemma3n manually then
---------
Signed-off-by: Tcc0403 <76503978+Tcc0403@users.noreply.github.com>
Co-authored-by: Raushan Turganbay <raushan@huggingface.co>