cleanup

Author: lucidrains

Diff for: vit_pytorch/simple_vit.py

@@ -1,7 +1,7 @@
 import torch
 from torch import nn
 
-from einops import rearrange, repeat
+from einops import rearrange
 from einops.layers.torch import Rearrange
 
 # helpers