Add types to vision_transformers.py

[Laurent2916]

Hi !

In continuation of #1989, here's a PR that adds the missing types to models/vision_transformers.py, plus a few other things.
I haven't added any docstrings, as VisionTransformer already has a docstring.

[Laurent2916]

Using einops here could improve readability, though transformers chose not to, as it may interfere with torch.compiler or onnx.export (see huggingface/transformers#25110, arogozhnikov/einops#250 and arogozhnikov/einops#274)

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[rwightman]

@Laurent2916 I do like the typing changes, but I do not want the rest (formatting, pos -> kwarg for obvious ones, inplace changes x +=, etc).

Few comments on formatting:

1. I disagree strongly with the black function arg alignment, it should have an extra indent level, I really hope ruff adds an option (Formatter: Specify indentation for function parameters astral-sh/ruff#8360) as black has refused despite it being mentioned in PEP and existing in IDE formatting rules for ages (like PyCharm that I use)
2. I've been writing Python for a long time and I have the old school single quote engrained. I'm not super attached to this one but really see the point in mass changing until I decide to adopt a formatter some day, haven't found one I like without inordinat config just yet

startswith has existested for > 20 years, I'm old. It is the remove functions that are new.

[Laurent2916]

Alright I reverted most of the formatting that I applied.
You're right about startswith, I didn't read the PEP carefully.

[rwightman]

@Laurent2916 awesome, thx, should be able to merge this later today