Fix BNN doc to work with Mooncake

TuringLang · Nov 25, 2024 · 02c24ef · 02c24ef
1 parent 28fcc61
commit 02c24ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/03-bayesian-neural-network/index.qmd b/tutorials/03-bayesian-neural-network/index.qmd
@@ -189,7 +189,7 @@ const nn = StatefulLuxLayer{true}(nn_initial, nothing, st)
     parameters ~ MvNormal(zeros(nparameters), Diagonal(abs2.(sigma .* ones(nparameters))))
 
     # Forward NN to make predictions
-    preds = Lux.apply(nn, xs, vector_to_parameters(parameters, ps))
+    preds = Lux.apply(nn, xs, f32(vector_to_parameters(parameters, ps)))
 
     # Observe each prediction.
     for i in eachindex(ts)