Commit
•
93821dc
1
Parent(s):
b971cbc
Fix Getting Started example (#5)
Browse files- Fix Getting Started example (f1c07ef8d08c0562cb4457644fb8ebdab27a7236)
Co-authored-by: Gerald Woo <[email protected]>
README.md
CHANGED
|
@@ -84,7 +84,7 @@ test_data = test_template.generate_instances(
|
|
| 84 |
)
|
| 85 |
|
| 86 |
# Prepare pre-trained model by downloading model weights from huggingface hub
|
| 87 |
-
model = MoiraiForecast
|
| 88 |
module=MoiraiModule.from_pretrained(f"Salesforce/moirai-1.0-R-{SIZE}"),
|
| 89 |
prediction_length=PDT,
|
| 90 |
context_length=CTX,
|
|
|
|
| 84 |
)
|
| 85 |
|
| 86 |
# Prepare pre-trained model by downloading model weights from huggingface hub
|
| 87 |
+
model = MoiraiForecast(
|
| 88 |
module=MoiraiModule.from_pretrained(f"Salesforce/moirai-1.0-R-{SIZE}"),
|
| 89 |
prediction_length=PDT,
|
| 90 |
context_length=CTX,
|