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,