Skip to content

Commit

Permalink
Update timm universal (support transformer-style model) (#1004)
Browse files Browse the repository at this point in the history
* Update timm_universal.py

* Fix ruff style and typing

* Update encoders_timm.rst

* Fix typo error

* Fix typo error & Update doc

* Fix typo error

* Support channel-last format

* Update encoders_timm.rst

* Update timm_universal.py

* Fix ruff style

* Update timm_universal.py

1. rename temporary model
2. create temporary model on meta device to speed up

* Add tests/test_models & fix type

* Update test_models.py

* Update test_models.py
  • Loading branch information
brianhou0208 authored Dec 19, 2024
1 parent 34ee31d commit aff25b1
Show file tree
Hide file tree
Showing 3 changed files with 879 additions and 155 deletions.
Loading

0 comments on commit aff25b1

Please sign in to comment.