gligen is missing math import

This commit is contained in:
doctorpangloss 2024-01-03 16:01:44 -08:00
parent 369aeb598f
commit d31298ac60

View File

@ -1,3 +1,5 @@
import math
import torch
from torch import nn, einsum
from .ldm.modules.attention import CrossAttention