Lesser print statements

model/patch_embed.py

@@ -82,9 +82,7 @@ class PatchEmbedding(nn.Module):
         grid_size_h = actual_h // self.patch_height
         grid_size_w = actual_w // self.patch_width
         
-        print(f"Grid size: {grid_size_h} x {grid_size_w} = {grid_size_h * grid_size_w}")
-        print(f"Expected patches: {out.shape[1]}")
-
+        
         out = self.patch_embed(out)
         pos_emb = get_patch_positional_embedding(
             self.hidden_dim, grid_size=(grid_size_h, grid_size_w), device=x.device