Add Time-LLM

.circleci/config.yml

@@ -8,7 +8,7 @@ jobs:
       - checkout
       - run:
           name: Install dependencies
-          command: micromamba install -n base -c conda-forge -y python=3.10 git -f environment.yml
+          command: micromamba install -n base -c conda-forge -y python=3.10 git -f environment-cpu.yml
       - run:
           name: Run nbdev tests
           command: |
@@ -24,7 +24,7 @@ jobs:
       - checkout
       - run:
           name: Install dependencies
-          command: micromamba install -n base -c conda-forge -y python=3.10 -f environment.yml
+          command: micromamba install -n base -c conda-forge -y python=3.10 -f environment-cpu.yml
       - run:
           name: Run model performance tests
           command: |
@@ -48,7 +48,7 @@ jobs:
       - checkout
       - run:
           name: Install dependencies
-          command: micromamba install -n base -c conda-forge -y python=3.10 -f environment.yml
+          command: micromamba install -n base -c conda-forge -y python=3.10 -f environment-cpu.yml
       - run:
           name: Run model performance tests
           command: |

nbs/core.ipynb

@@ -86,8 +86,7 @@
     "    MLP, NHITS, NBEATS, NBEATSx, DLinear, NLinear,\n",
     "    TFT, VanillaTransformer,\n",
     "    Informer, Autoformer, FEDformer,\n",
-    "    StemGNN, PatchTST, TimesNet,\n",
-    "    TSMixer,\n",
+    "    StemGNN, PatchTST, TimesNet, TimeLLM, TSMixer\n",
     ")"
    ]
   },
@@ -224,6 +223,7 @@
     "    'tft': TFT, 'autotft': TFT,\n",
     "    'timesnet': TimesNet, 'autotimesnet': TimesNet,\n",
     "    'vanillatransformer': VanillaTransformer, 'autovanillatransformer': VanillaTransformer,\n",
+    "    'timellm': TimeLLM,\n",
     "    'tsmixer': TSMixer, 'autotsmixer': TSMixer,\n",
     "}"
    ]