ACTUALLY fix the ruff check error with unused import

2025-11-14 12:09:08 +01:00 · 2025-11-14 12:09:08 +01:00 · b9be0b76b2
parent c4ebef5882
commit b9be0b76b2
2 changed files with 2 additions and 2 deletions
--- a/comfy/ldm/common_dit.py
+++ b/comfy/ldm/common_dit.py
@ -1,5 +1,5 @@
 import torch
-from einops import rearrange
+from einops import rearrange, repeat
 import comfy.rmsnorm


--- a/comfy/ldm/flux/model.py
+++ b/comfy/ldm/flux/model.py
@ -4,7 +4,7 @@ from dataclasses import dataclass

 import torch
 from torch import Tensor, nn
-from einops import rearrange, repeat
+from einops import rearrange
 import comfy.ldm.common_dit
 import comfy.patcher_extension