From eb45a7924ef1f9d88296a99521103e9f1d4be6ed Mon Sep 17 00:00:00 2001
From: benjijamorris <54606172+benjijamorris@users.noreply.github.com>
Date: Thu, 29 Feb 2024 13:55:55 -0800
Subject: [PATCH] fix import (#341)

---
 cyto_dl/nn/vits/mae.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/cyto_dl/nn/vits/mae.py b/cyto_dl/nn/vits/mae.py
index 95e4dff7b..61d8f336c 100644
--- a/cyto_dl/nn/vits/mae.py
+++ b/cyto_dl/nn/vits/mae.py
@@ -11,7 +11,6 @@
 from timm.models.layers import trunc_normal_
 from timm.models.vision_transformer import Block
 
-from cyto_dl.nn.vits.blocks.attention_autoencoder import AttentionAutoencoder
 from cyto_dl.nn.vits.cross_mae import CrossMAE_Decoder