Fixes typo in toy experiment configuration.

PiperOrigin-RevId: 350170372
Change-Id: I01c04264b47863f96fa45e2f51ebb3573fa4e69f

Author: Johannes Ballé

models/toy_sources/toy_sources.ipynb

@@ -794,7 +794,7 @@
         "validation_batch_size = 65536\n",
         "learning_rate = 1e-3\n",
         "\n",
-        "latent_dims = 1\n",
+        "latent_dims = 2\n",
         "analysis_filters = [100, 100]\n",
         "analysis_activation = \"softplus\"\n",
         "synthesis_filters = [100, 100]\n",