faster decoding related changes

2025-10-01 16:50:27 +00:00
parent 2fded92687
commit a1c61b6f5d
5 changed files with 228 additions and 149 deletions
@@ -57,8 +57,12 @@ pip install -e .
 ### Code Example

 ```python
+import torch
 import numpy as np
 import timesfm
+
+torch.set_float32_matmul_precision("high")
+
 model = TimesFM_2p5_200M_torch.from_pretrained("google/timesfm-2.5-200m-pytorch")

 model.compile(