freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor changes to readme

Browse Source
This commit is contained in:
Rajat Sen
2025-10-01 16:51:59 +00:00
parent a1c61b6f5d d5a3d5b872
commit 2b5516d2f1
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2 -1
View File
@@ -63,7 +63,7 @@ import timesfm
torch.set_float32_matmul_precision("high") torch.set_float32_matmul_precision("high")
model = TimesFM_2p5_200M_torch.from_pretrained("google/timesfm-2.5-200m-pytorch") model = timesfm.TimesFM_2p5_200M_torch.from_pretrained("google/timesfm-2.5-200m-pytorch")
model.compile( model.compile(
timesfm.ForecastConfig( timesfm.ForecastConfig(
@@ -74,6 +74,7 @@ model.compile(
force_flip_invariance=True, force_flip_invariance=True,
infer_is_positive=True, infer_is_positive=True,
fix_quantile_crossing=True, fix_quantile_crossing=True,
torch_compile=True,
) )
) )
point_forecast, quantile_forecast = model.forecast( point_forecast, quantile_forecast = model.forecast(