flax

3dfeff9f - Update 05_vae.py

Update 05_vae.py Optimzier alaready was instantiated using the model as a parameter, and the update function if it ever took 3 parameters it now takes two (including self) hence one gets an error and this code does not run.