Add num_heads variables

Make changes

Resolve run time errors

Change any output errors

Add AutoEncoder (VQVAE)

Add UpBlock Unet to concat downblock's output

Add all the layers in the UpBlock

Add forward function to the midblock

Add residual convolution to the mid block

Add MidBlocks and change cross_attn in down blocks

Add cross attn if needed for conditional latent diffusion

Add forward function with attention

Add cross attention and down sample layers

Add attention to Down Blocks