diff --git a/comfy/ldm/chroma/layers.py b/comfy/ldm/chroma/layers.py
index 9f4ad5bd2158..2d568434871b 100644
--- a/comfy/ldm/chroma/layers.py
+++ b/comfy/ldm/chroma/layers.py
@@ -7,6 +7,9 @@
     ModulationOut,
 )
 
+# TODO: remove this in a few months
+SingleStreamBlock = None
+DoubleStreamBlock = None
 
 
 class ChromaModulationOut(ModulationOut):