Commit
•
a41338c
1
Parent(s):
7ab54e2
Fix Getting Started example (#5)
Browse files- Fix Getting Started example (7ee568f82cfef5709a36339573fa30550a8183b2)
Co-authored-by: Gerald Woo <gorold@users.noreply.huggingface.co>
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,
|