Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ from transformers import AutoModelForCausalLM
|
|
| 59 |
# load pretrain model
|
| 60 |
# supports different lookback/forecast lengths
|
| 61 |
model = AutoModelForCausalLM.from_pretrained(
|
| 62 |
-
'
|
| 63 |
trust_remote_code=True,
|
| 64 |
device_map="auto"
|
| 65 |
)
|
|
|
|
| 59 |
# load pretrain model
|
| 60 |
# supports different lookback/forecast lengths
|
| 61 |
model = AutoModelForCausalLM.from_pretrained(
|
| 62 |
+
'thuml/Timer-S1',
|
| 63 |
trust_remote_code=True,
|
| 64 |
device_map="auto"
|
| 65 |
)
|