Fix custom nodes import error. (comfyanonymous#10747)

comfyanonymous · web-flow · commit 443056c401c5 · 2025-11-14T03:26:05.000-05:00
This should fix the import errors but will break if the custom nodes actually try to use the class.
diff --git 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):

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,9 @@`
`7`	`7`	`ModulationOut,`
`8`	`8`	`)`
`9`	`9`
	`10`	`+# TODO: remove this in a few months`
	`11`	`+SingleStreamBlock = None`
	`12`	`+DoubleStreamBlock = None`
`10`	`13`
`11`	`14`
`12`	`15`	`class ChromaModulationOut(ModulationOut):`