diff --git a/comfy/ldm/chroma_radiance/model.py b/comfy/ldm/chroma_radiance/model.py
index 4339f457e..5941351de 100644
--- a/comfy/ldm/chroma_radiance/model.py
+++ b/comfy/ldm/chroma_radiance/model.py
@@ -7,7 +7,7 @@ from typing import Optional
 
 import torch
 from torch import Tensor, nn
-from einops import rearrange, repeat
+from einops import repeat
 import comfy.ldm.common_dit
 import comfy.patcher_extension