Fix Getting Started example
#5
by gorold - opened
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,
|