diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py
index 21004d4946..f83a1753da 100755
--- a/src/diffusers/models/attention_processor.py
+++ b/src/diffusers/models/attention_processor.py
@@ -11,6 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from __future__ import annotations
+
 import inspect
 import math
 from typing import Callable