diff --git a/bambi/models.py b/bambi/models.py index ed6e11104..51f250664 100644 --- a/bambi/models.py +++ b/bambi/models.py @@ -277,6 +277,7 @@ def fit( chains=chains, cores=cores, random_seed=random_seed, + vi_kwargs=vi_kwargs, **kwargs, )