diff --git a/comfy/ldm/chroma/layers.py b/comfy/ldm/chroma/layers.py
index 410b7121f..dd0b72f70 100644
--- a/comfy/ldm/chroma/layers.py
+++ b/comfy/ldm/chroma/layers.py
@@ -1,10 +1,7 @@
-import math
-from dataclasses import dataclass
-
 import torch
 from torch import Tensor, nn
 
-from .math import attention, rope
+from .math import attention
 from comfy.ldm.flux.layers import (
     MLPEmbedder,
     RMSNorm,
@@ -12,7 +9,6 @@ from comfy.ldm.flux.layers import (
     SelfAttention,
     ModulationOut,
 )
-import comfy.ldm.common_dit