use static cache if has layer_types but no sliding_window

Author: guangy10

src/transformers/integrations/executorch.py

@@ -56,19 +56,13 @@ def __init__(
         if not hasattr(model.config, "use_cache") or model.config.use_cache is False:
             raise ValueError("The model must have caching enabled to be performant.")
 
-        if not hasattr(model.config, "cache_implementation"):
-            # If `cache_implementation` is not specified explicitly in the config, `DynamicCache` will
-            # be used by default, so export will use `StaticCache` by default.
-            logging.info("Using `StaticCache` for export as `cache_implementation` is not specified in the config.")
+        if not hasattr(model.config, "layer_types") and model.config.get("sliding_window", None) is not None:
+            # If `layer_types` is not specified explicitly in the config, there is only 1 type of layers, so
+            # export will use `StaticCache` by default.
+            logging.info("Using `StaticCache` for export as `layer_types` is not specified in the config.")
             self.model = TorchExportableModuleWithStaticCache(model)
         else:
-            if model.config.cache_implementation == "hybrid":
-                self.model = TorchExportableModuleWithHybridCache(model, max_batch_size, max_cache_len)
-            else:
-                raise ValueError(
-                    f"Unsupported cache implementation: {model.config.cache_implementation}. "
-                    "Please use `hybrid` or `static`."
-                )
+            self.model = TorchExportableModuleWithHybridCache(model, max_batch_size, max_cache_len)
 
     def forward(
         self,
@@ -406,12 +400,6 @@ def __init__(
         if not self.model.config.use_cache:
             raise AssertionError("Model must have caching enabled")
 
-        if (
-            not hasattr(self.model.config, "cache_implementation")
-            or self.model.config.cache_implementation != "hybrid"
-        ):
-            raise AssertionError("Model must use 'hybrid' cache implementation")
-
         # Initialize the HybridCache
         self.cache = HybridCache(
             config=self.model.config,