diff --git a/comfy/ldm/lumina/model.py b/comfy/ldm/lumina/model.py
index 24c6d80f2..4eb7164d1 100644
--- a/comfy/ldm/lumina/model.py
+++ b/comfy/ldm/lumina/model.py
@@ -1,4 +1,5 @@
 # Code from: https://github.com/Alpha-VLLM/Lumina-Image-2.0/blob/main/models/model.py
+from __future__ import annotations
 
 from typing import List, Optional, Tuple