Commit
•
a41338c
1
Parent(s):
7ab54e2
Fix Getting Started example (#5)
Browse files- Fix Getting Started example (7ee568f82cfef5709a36339573fa30550a8183b2)
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,
|