bayesflow-org
diff --git a/‎bayesflow/networks/diffusion_model/__init__.py‎
Lines changed: 1 addition & 0 deletions b/‎bayesflow/networks/diffusion_model/__init__.py‎
Lines changed: 1 addition & 0 deletions
@@ -1,4 +1,5 @@
 from .diffusion_model import DiffusionModel
+from .compositional_diffusion_model import CompositionalDiffusionModel
 from .schedules import CosineNoiseSchedule
 from .schedules import EDMNoiseSchedule
 from .schedules import NoiseSchedule
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from .diffusion_model import DiffusionModel`
	`2`	`+from .compositional_diffusion_model import CompositionalDiffusionModel`
`2`	`3`	`from .schedules import CosineNoiseSchedule`
`3`	`4`	`from .schedules import EDMNoiseSchedule`
`4`	`5`	`from .schedules import NoiseSchedule`