diff --git a/src/ConformalGenerator.jl b/src/ConformalGenerator.jl
index 540d71a6b64d446bcbdbb219cdca1aaeaee89a80..bde032bd2aea466c0b1729d09823b37755b60ab1 100644
--- a/src/ConformalGenerator.jl
+++ b/src/ConformalGenerator.jl
@@ -18,6 +18,8 @@ end
 @with_kw struct ConformalGeneratorParams
     opt::Flux.Optimise.AbstractOptimiser = Descent()
     Ï„::AbstractFloat = 1e-3
+    κ::Real = 1.0
+    Τ::Real = 0.5
 end
 
 """