Add Time-LLM (#908)

Nixtla · Mar 8, 2024 · 076f863 · 076f863
1 parent 906ea9f
commit 076f863
Show file tree

Hide file tree

Showing 9 changed files with 1,556 additions and 8 deletions.
diff --git a/.circleci/config.yml b/.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: |

diff --git a/nbs/core.ipynb b/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",
     "}"
    ]

diff --git a/nbs/imgs_models/timellm.png b/nbs/imgs_models/timellm.png