Add dpt-hybrid support (#20645)
* add `dpt-hybrid` support * refactor * final changes, all tests pass * final cleanups * final changes * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * fix docstring * fix typo * change `vit_hybrid` to `hybrid` * replace dataclass * add docstring * move dataclasses * fix test * add `PretrainedConfig` support for `backbone_config` * fix docstring * Apply suggestions from code review Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * remove `embedding_type` and replace it by `is_hybrid` Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
This commit is contained in:
@@ -376,6 +376,10 @@ SPECIAL_MODULE_TO_TEST_MAP = {
|
||||
"models/gpt2/test_modeling_gpt2.py",
|
||||
"models/megatron_gpt2/test_modeling_megatron_gpt2.py",
|
||||
],
|
||||
"models/dpt/modeling_dpt.py": [
|
||||
"models/dpt/test_modeling_dpt.py",
|
||||
"models/dpt/test_modeling_dpt_hybrid.py",
|
||||
],
|
||||
"optimization.py": "optimization/test_optimization.py",
|
||||
"optimization_tf.py": "optimization/test_optimization_tf.py",
|
||||
"pipelines/__init__.py": "pipelines/test_pipelines_*.py",
|
||||
|
||||
Reference in New Issue
Block a user