Skip to content

Commit 6d4961e

Browse files
authored
Tweak for compat
1 parent c032366 commit 6d4961e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/processes.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ function NoisyInterpolatingDiscreteFlow(noise; K = 1, dummy_token::T = nothing)
6565
)
6666
end
6767
NoisyInterpolatingDiscreteFlow() = NoisyInterpolatingDiscreteFlow{Nothing}(0.2)
68+
NoisyInterpolatingDiscreteFlow(noise, power) = NoisyInterpolatingDiscreteFlow(noise, K = power)
6869
function bridge(p::NoisyInterpolatingDiscreteFlow, x0::DiscreteState{<:AbstractArray{<:Signed}}, x1::DiscreteState{<:AbstractArray{<:Signed}}, t)
6970
D = size(x0.state)
7071
ts = expand(t, ndims(x0.state))
@@ -127,4 +128,4 @@ function step(P::NoisyInterpolatingDiscreteFlow{<:Integer}, Xₜ::DiscreteState{
127128
newXₜ = CategoricalLikelihood(eltype(s₁).(tensor(ohXₜ) .+ (Δt .* velo)))
128129
clamp!(tensor(newXₜ), 0, Inf)
129130
return rand(newXₜ)
130-
end
131+
end

0 commit comments

Comments
 (0)