XSF0528 commited on
Commit
4da4d58
·
verified ·
1 Parent(s): e0f11a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ pip install torch transformers einops numpy
38
  import torch
39
  from transformers import AutoModel
40
 
41
- model = AutoModel.from_pretrained("<namespace>/Tabby", trust_remote_code=True).eval()
42
 
43
  history = torch.randn(4, 8096).cumsum(-1) # [B, Tc] — any length; NaN = unobserved
44
  q = model.predict(history, prediction_length=96) # [4, 99, 96]
 
38
  import torch
39
  from transformers import AutoModel
40
 
41
+ model = AutoModel.from_pretrained("paris-noah/Tabby", trust_remote_code=True).eval()
42
 
43
  history = torch.randn(4, 8096).cumsum(-1) # [B, Tc] — any length; NaN = unobserved
44
  q = model.predict(history, prediction_length=96) # [4, 99, 96]