Modular: support for importing functions from any file (#35692)

* fix function imports

* improve comment

* Update modeling_switch_function.py

* make checks more robust

* improvement

* rename

* final test update
This commit is contained in:
Cyril Vallez
2025-01-16 16:37:53 +00:00
committed by GitHub
parent 8ebe9d7166
commit 91be6a5eb2
10 changed files with 305 additions and 43 deletions

View File

@@ -107,7 +107,6 @@ class NewTaskModelPreTrainedModel(PreTrainedModel):
_supports_cache_class = True
_supports_quantized_cache = True
_supports_static_cache = True
_supports_cache_class = True
_supports_flash_attn_2 = True
_supports_sdpa = True