From a246d2bb64e91bb2a9d7ab1522f19e50405bacbd Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Sun, 28 Apr 2024 01:01:31 +0700
Subject: [PATCH] try sep vit

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

diff --git a/models.py b/models.py
index ccb1ce9..f0b6d18 100644
--- a/models.py
+++ b/models.py
@@ -1,5 +1,5 @@
 import torch
-from torch import nn
+from torch import nn, einsum
 import torch.nn.functional as F
 import numpy as np
 from functools import partial