diff --git a/comfy/ldm/common_dit.py b/comfy/ldm/common_dit.py index a31c9f571..3f6f38917 100644 --- a/comfy/ldm/common_dit.py +++ b/comfy/ldm/common_dit.py @@ -1,5 +1,5 @@ import torch -from einops import rearrange, repeat +from einops import rearrange import comfy.rmsnorm