Make ViTPooler configurable #36517
Cyrilvallez
merged
31 commits into
huggingface:main
from
sebbaur:feature/vit_pooler_configurable
Make ViT Pooler configurable, so that it is possible to pick the acti…
97199e31
Add documentation and allow functions as activations (instead of just…
43639fa6
formatting change
80e132ae
Use ACT2FN
9fe06915
Formatting change
bd2a0011
Formatting changes
18edafb6
qubvel
commented
on 2025-03-03
force pooler_act to be string
5060f56b
force pooler_act to be string
0a2895f2
Add configs to OBJECTS_TO_IGNORE to make check_docstrings happy
9e9dfd1f
Making the same change in ijepa to make check_modular_conversion happy
0d2b485f
Add IJepaConfig to make CI happy
e0d6928d
rename pooler_size to pooler_output_size as defined in the config
0881e5f5
typo
e2670cca
Merge branch 'main' into feature/vit_pooler_configurable
03829602
qubvel
commented
on 2025-03-04
revert change to ignore variable
c89f69b7
Merge branch 'feature/vit_pooler_configurable' of https://github.com/…
6309d96d
Merge branch 'main' into feature/vit_pooler_configurable
6c6a1ac9
Merge branch 'main' into feature/vit_pooler_configurable
5ffac413
Ran utils/check_docstrings.py --fix_and_overwrite
ad2aecad
revert unrelated change
bc5ef1b5
remove redundant defaults
2d0071f5
rename self.act -> self.activation
3ca67c1b
Merge branch 'main' into feature/vit_pooler_configurable
05499eb0
Merge branch 'main' into feature/vit_pooler_configurable
ab56b63c
Merge branch 'main' into feature/vit_pooler_configurable
308b5b3d
Merge branch 'main' into feature/vit_pooler_configurable
7e9bc63b
Merge branch 'main' into feature/vit_pooler_configurable
eff0bd13
Merge branch 'main' into feature/vit_pooler_configurable
8028f626
tanh activation function in mapping
87b9162e
Merge branch 'feature/vit_pooler_configurable' of https://github.com/…
16dfb96e
Merge branch 'main' into feature/vit_pooler_configurable
366edfed
Login to write a write a comment.
Login via GitHub