Merge 1bb3f8a into 4affc28

TuringLang · Aug 24, 2023 · 4d52d0f · 4d52d0f
2 parents 4affc28 + 1bb3f8a
commit 4d52d0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/variational/advi.jl b/src/variational/advi.jl
@@ -75,7 +75,7 @@ function Bijectors.bijector(
         if d isa Distributions.UnivariateDistribution
             b
         else
-            Vec(b, size(d))
+            Vec(b, size(Bijectors.transformed(d)))
         end
     end