doyensahoo gorold commited on
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]>

Files changed (1) hide show
  1. README.md +1 -1
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.load_from_checkpoint(
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,