diff --git a/comfy/gligen.py b/comfy/gligen.py
index 8d182839e..f47baa0f1 100644
--- a/comfy/gligen.py
+++ b/comfy/gligen.py
@@ -1,3 +1,5 @@
+import math
+
 import torch
 from torch import nn, einsum
 from .ldm.modules.attention import CrossAttention