From 23c44d35825f9d92f58b77b72dae068e1e3e9fee Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Sun, 28 Apr 2024 11:59:09 +0700
Subject: [PATCH] try gtp vit

---
 models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/models.py b/models.py
index 947ce5a..60e6eb8 100644
--- a/models.py
+++ b/models.py
@@ -10,6 +10,7 @@ from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 from timm.models.layers import DropPath, trunc_normal_
 from timm.models.registry import register_model
 from timm.layers.helpers import to_2tuple
+from typing import *
 
 
 class ConvE(torch.nn.Module):