update review

Author: duyifanict

baselines/TimeMoE/config/timemoe_large.py

@@ -21,7 +21,7 @@
 
 MODEL_PARAM = {
     'model_id': "baselines/TimeMoE/ckpt/TimeMoE-200M",
-    'from_pretrained': False,
+    'from_pretrained': pretrained,
     'context_length': 4095,
     'trust_remote_code': True,
 }

basicts/runners/base_utsf_runner.py

@@ -22,8 +22,8 @@ class BaseUniversalTimeSeriesForecastingRunner(BaseIterationRunner):
     def __init__(self, cfg: Dict):
         super().__init__(cfg)
 
-        # setup graph flag
-        self.need_setup_graph = cfg['MODEL'].get('SETUP_GRAPH', False)
+        # setup graph flag, utsf models do not need
+        self.need_setup_graph = False
 
         # initialize scaler
         self.data_scaler = self.build_data_scaler(cfg)

server/engine/engine.py

@@ -9,6 +9,8 @@
 from easytorch.utils import set_visible_devices
 
 sys.path.append(os.path.abspath(os.path.join(__file__,'..','..','..')))
+
+
 class inference_engine(object):
     """
     Inference engine for EasyTorch.