Sloving conflicts

Re training will finish this time for sure

Halfway through

Modify training code

Remove logging statements

Print statements for logging

Add num_heads variables

Make changes

Resolve run time errors

Get the buffers in order to open as image

Change any output errors

Move the pretrained vgg16 model to another directory

Add forward function to the Diffusion model UNet

Build the Unet from the config

change variable names

Get all parameters from model_config

Add discriminator for adverserial loss for VAE training

Add lpips for VAE training

Clean up

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