diff --git a/exp/exp_main.py b/exp/exp_main.py
index db79626..839f93e 100644
--- a/exp/exp_main.py
+++ b/exp/exp_main.py
@@ -1,4 +1,4 @@
-from models.ETSformer.model import ETSformer
+from models import ETSformer
 from data_provider.data_factory import data_provider
 from exp.exp_basic import Exp_Basic
 from utils.tools import EarlyStopping, adjust_learning_rate
@@ -210,8 +210,8 @@ class Exp_Main(Exp_Basic):
                 outputs = outputs.detach().cpu().numpy()
                 batch_y = batch_y.detach().cpu().numpy()
 
-                pred = outputs  # outputs.detach().cpu().numpy()  # .squeeze()
-                true = batch_y  # batch_y.detach().cpu().numpy()  # .squeeze()
+                pred = outputs
+                true = batch_y
 
                 preds.append(pred)
                 trues.append(true)
diff --git a/models/__init__.py b/models/__init__.py
index e69de29..adbf454 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -0,0 +1 @@
+from .etsformer.model import ETSformer
diff --git a/models/ETSformer/__init__.py b/models/etsformer/__init__.py
similarity index 100%
rename from models/ETSformer/__init__.py
rename to models/etsformer/__init__.py
diff --git a/models/ETSformer/decoder.py b/models/etsformer/decoder.py
similarity index 100%
rename from models/ETSformer/decoder.py
rename to models/etsformer/decoder.py
diff --git a/models/ETSformer/encoder.py b/models/etsformer/encoder.py
similarity index 100%
rename from models/ETSformer/encoder.py
rename to models/etsformer/encoder.py
diff --git a/models/ETSformer/exponential_smoothing.py b/models/etsformer/exponential_smoothing.py
similarity index 100%
rename from models/ETSformer/exponential_smoothing.py
rename to models/etsformer/exponential_smoothing.py
diff --git a/models/ETSformer/model.py b/models/etsformer/model.py
similarity index 100%
rename from models/ETSformer/model.py
rename to models/etsformer/model.py
diff --git a/models/ETSformer/modules.py b/models/etsformer/modules.py
similarity index 100%
rename from models/ETSformer/modules.py
rename to models/etsformer/modules.py