diff --git a/src/transformers/integrations/executorch.py b/src/transformers/integrations/executorch.py index 5fdde975914..171522bf75f 100644 --- a/src/transformers/integrations/executorch.py +++ b/src/transformers/integrations/executorch.py @@ -314,6 +314,7 @@ class TorchExportableModuleWithStaticCache(torch.nn.Module): kwargs = { "input_ids": input_ids, "position_ids": position_ids, + "attention_mask": None, "cache_position": cache_position, "past_key_values": past_key_values, "use_cache": True,