cleanup

lucidrains · web-flow · commit ddff7a7ea294 · 2022-05-03T20:23:16.000-07:00
diff --git a/vit_pytorch/simple_vit.py b/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
