transformers

Fix `layer_types` type hint for `AFMoE` and `Llama4`

#44874

Merged