cleanup

lucidrains · May 4, 2022 · ddff7a7 · ddff7a7
1 parent ef8c0ac
commit ddff7a7
Showing 1 changed file with 1 addition and 1 deletion.
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