Yong99 commited on
Commit
89e7340
·
verified ·
1 Parent(s): 1f28be3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -34,7 +34,7 @@ import torch
34
  from transformers import AutoModelForCausalLM
35
 
36
  # load pretrain model
37
- model = AutoModelForCausalLM.from_pretrained('thuml/timer-base', trust_remote_code=True, token=True)
38
 
39
  # prepare input
40
  batch_size, lookback_length = 1, 2880
 
34
  from transformers import AutoModelForCausalLM
35
 
36
  # load pretrain model
37
+ model = AutoModelForCausalLM.from_pretrained('thuml/timer-1.1-84m', trust_remote_code=True, token=True)
38
 
39
  # prepare input
40
  batch_size, lookback_length = 1, 2880