Update README.md
Browse files
README.md
CHANGED
|
@@ -46,8 +46,8 @@ pip install -e .
|
|
| 46 |
```python
|
| 47 |
import numpy as np
|
| 48 |
import timesfm
|
| 49 |
-
model = timesfm.TimesFM_2p5_200M_torch()
|
| 50 |
-
|
| 51 |
model.compile(
|
| 52 |
timesfm.ForecastConfig(
|
| 53 |
max_context=1024,
|
|
|
|
| 46 |
```python
|
| 47 |
import numpy as np
|
| 48 |
import timesfm
|
| 49 |
+
model = timesfm.TimesFM_2p5_200M_torch.from_pretrained("google/timesfm-2.5-200m-pytorch")
|
| 50 |
+
|
| 51 |
model.compile(
|
| 52 |
timesfm.ForecastConfig(
|
| 53 |
max_context=1024,
|