hf path

ibm-granite · Aug 30, 2024 · a831b95 · a831b95
1 parent da42a85
commit a831b95
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb b/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb
@@ -969,7 +969,7 @@
    "source": [
     "# Load model\n",
     "model = TinyTimeMixerForPrediction.from_pretrained(\n",
-    "    \"ibm/TTM\", revision=\"main\", prediction_filter_length=M4_FORECAST_HORIZON\n",
+    "    \"ibm-granite/granite-timeseries-ttm-v1\", revision=\"main\", prediction_filter_length=M4_FORECAST_HORIZON\n",
     ").to(device)"
    ]
   },
@@ -1074,7 +1074,7 @@
    "outputs": [],
    "source": [
     "model = TinyTimeMixerForPrediction.from_pretrained(\n",
-    "    \"ibm/TTM\",\n",
+    "    \"ibm-granite/granite-timeseries-ttm-v1\",\n",
     "    revision=\"main\",\n",
     "    prediction_filter_length=M4_FORECAST_HORIZON,\n",
     "    head_dropout=0.0,\n",
@@ -1507,7 +1507,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.12"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,