Update models/vae.py

models/vae.py

@@ -4,7 +4,7 @@ import pytorch_lightning as pl
 class VAEModel(pl.LightningModule):
     def __init__(self):
         super().__init__()
-        # Your VAE implementation here
+        # VAE implementation
         self.encoder = torch.nn.Sequential(
             torch.nn.Conv2d(1, 16, kernel_size=3, stride=2, padding=1),
             torch.nn.ReLU()