diff --git a/modules/intel/ipex/attention.py b/modules/intel/ipex/attention.py
index 377d5febd..c3e8ef8ed 100644
--- a/modules/intel/ipex/attention.py
+++ b/modules/intel/ipex/attention.py
@@ -1,3 +1,5 @@
+from typing import Tuple, Optional
+
 import os
 import torch
 from functools import cache, wraps