add optional time embedding layer to consistency model

Open arrjon opened this issue 4 weeks ago • 1 comments

This pull request introduces a new option to provide a custom time embedding layer to the StableConsistencyModel class, improving its flexibility and configurability. This also fixes a bug while loading a non-default FourierEmbedding.

Dec 23 '25 12:12 arrjon

Codecov Report

:x: Patch coverage is 0% with 3 lines in your changes missing coverage. Please review.

Files with missing lines	Patch %	Lines
...able_consistency_model/stable_consistency_model.py	0.00%	3 Missing :warning:

Files with missing lines	Coverage Δ
...able_consistency_model/stable_consistency_model.py	`20.80% <0.00%> (-0.34%)`	:arrow_down:

Dec 23 '25 13:12 codecov[bot]