Skip to content

Commit 4f43867

Browse files
committed
Move continuous-time consistency model to experimental module.
Related to #256 As we have not extensively tested their behavior and I have observed subpar performance in some settings, we move this model class to experimental for further testing, to see whether it is ready for broader use, or whether it turns out to be too unstable to be practical.
1 parent 68a2b47 commit 4f43867

File tree

5 files changed

+104
-110
lines changed

5 files changed

+104
-110
lines changed

bayesflow/experimental/__init__.py

Whitespace-only changes.

bayesflow/networks/consistency_models/continuous_consistency_model.py renamed to bayesflow/experimental/continuous_time_consistency_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,12 @@
1919
)
2020

2121

22-
from ..inference_network import InferenceNetwork
23-
from ..embeddings import FourierEmbedding
22+
from bayesflow.networks import InferenceNetwork
23+
from bayesflow.networks.embeddings import FourierEmbedding
2424

2525

2626
@register_keras_serializable(package="bayesflow.networks")
27-
class ContinuousConsistencyModel(InferenceNetwork):
27+
class ContinuousTimeConsistencyModel(InferenceNetwork):
2828
"""Implements an sCM (simple, stable, and scalable Consistency Model)
2929
with continous-time Consistency Training (CT) as described in [1].
3030
The sampling procedure is taken from [2].

bayesflow/networks/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from .cif import CIF
2-
from .consistency_models import ConsistencyModel, ContinuousConsistencyModel
2+
from .consistency_models import ConsistencyModel
33
from .coupling_flow import CouplingFlow
44
from .deep_set import DeepSet
55
from .flow_matching import FlowMatching
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1 @@
11
from .consistency_model import ConsistencyModel
2-
from .continuous_consistency_model import ContinuousConsistencyModel

examples/Continuous_Consistency_Model_Playground.ipynb

Lines changed: 100 additions & 105 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)