rajatsen91 commited on
Commit
a6e2c7c
·
verified ·
1 Parent(s): ef46453

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
- model.load_checkpoint()
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,