Merge pull request #8 from kashif/hub

[Hub] Download weights automatically from hub
2024-05-10 10:09:31 -07:00
parent e2a527425b 2a83a0f82d
commit 2fafb794fa
2 changed files with 12 additions and 4 deletions
@@ -78,7 +78,7 @@ tfm = timesfm.TimesFm(
    model_dims=1280,
    backend=<backend>,
 )
-tfm.load_from_checkpoint(<checkpoint_path>)
+tfm.load_from_checkpoint(repo_id="google/timesfm-1.0-200m")
 ```

 Note that the four parameters are fixed to load the 200m model