Add samples at all good, less budget

Epoch 14 DIT

Epoch 13 DIT

Epoch 12 DIT

Epoch 10 DIT

Epoch 9 DIT

Epoch 8 DIT

Epoch 7 DIT

Epoch 6 DIT

Epoch 5 DIT

Epoch 3 DIT

Epoch 2 DIT

1st epoch DIT

Change the file name

Remove print statements

Remove print statements

Lesser print statements

Changes to the transformer layer

Made all changes needed to the patch embed layer for dimension mismatch

Add hidden_dim to patch embeddings

Changes in Dit block

Changes in the patch_embed

Add the setup_script

Add wandb logs to gitignore

Complete the script with wandb logging

Add the training script for dit

3rd Epoch of VAE

2nd epoch VAE

Update to 10 epochs

Add the dataset and the training script

Add the lpip model

Add the dataset

Add part one

Add the model and scheduler

initial commit