transformers/tests/models/hiera
Joshua Lochner c6d2848a23
🚨 Fix torch.jit.trace for interpolate_pos_encoding in all vision models (#33226)
* Fix `torch.jit.tracing` for `interpolate_pos_encoding` in all vision models

* Apply formatting

* Add missing `self.config = config`

* Fix copies

* Fix hiera interpolation unit test

* Formatting

* Update `_import_structure`

* make style

* Fix docstring

* Use `# Copied from` instead of utils

* DeiT variable renaming (`class_and_dist_pos_embed`)

* Fix Hiera `interpolate_pos_encoding`
2024-09-05 16:17:34 +02:00
..
__init__.py Adding hiera (#30356) 2024-07-11 22:13:56 +01:00
test_modeling_hiera.py 🚨 Fix torch.jit.trace for interpolate_pos_encoding in all vision models (#33226) 2024-09-05 16:17:34 +02:00