From 42995b86f26e91f06c5e82e7d6b8a36372da1805 Mon Sep 17 00:00:00 2001
From: wjj19950828 <wjjisloser@163.com>
Date: Wed, 12 Apr 2023 14:13:18 +0800
Subject: [PATCH] fixed import models bug

---
 x2paddle/project_convertor/pytorch/torch2paddle/varbase.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py b/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py
index b42648e..6edce6b 100644
--- a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py
+++ b/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py
@@ -53,6 +53,8 @@ def is_condition_two(idx):
     return False
 
 
+VarBase = core.eager.Tensor
+
 VarBase.tmp = VarBase.__getitem__
 
 
@@ -95,9 +97,8 @@ def __getitem__(self, idx):
             return out
 
 
-VarBase = core.eager.Tensor
-
 VarBase.__getitem__ = __getitem__
+
 VarBase.setitem_tmp = VarBase.__setitem__
 
 
-- 
GitLab