diff --git a/vit_pytorch/mae.py b/vit_pytorch/mae.py
index 41293ff..2afcfd5 100644
--- a/vit_pytorch/mae.py
+++ b/vit_pytorch/mae.py
@@ -1,8 +1,7 @@
 import torch
-from math import ceil
 from torch import nn
 import torch.nn.functional as F
-from einops import rearrange, repeat
+from einops import repeat
 
 from vit_pytorch.vit import Transformer