Commit
•
01d631d
1
Parent(s):
d41a1e5
Fix Getting Started example (#3)
Browse files- Fix Getting Started example (1d2eaddd22cc6bd56466bbcc458737329b1fe45f)
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,
|