From a831b9575b5e324db09090aca3381e7b06a7fa5d Mon Sep 17 00:00:00 2001 From: Wesley Gifford <79663411+wgifford@users.noreply.github.com> Date: Fri, 30 Aug 2024 09:25:38 -0400 Subject: [PATCH] hf path --- notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb b/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb index bdd9130c..ad2e767a 100644 --- 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,