Remove print statements

Lesser print statements

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

Add hidden_dim to patch embeddings

Changes in the patch_embed

Add the model and scheduler