Ccloud0525 commited on
Commit
7fc13de
·
verified ·
1 Parent(s): 1dde777

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -114,11 +114,11 @@ num_samples = 100
114
  # For inference_token_len, you can refer to LightGTS (Periodic Patching).
115
  # We recommend to use the period length as the inference_token_len.
116
  output = model.generate(inputs=batch_x,text_input_ids=batch_input_ids,
117
- text_attention_mask=batch_attention_mask,
118
  text_token_type_ids=batch_token_type_ids,
119
- max_output_length=forecast_length,
120
- num_samples=num_samples,
121
- inference_token_len=48)
122
 
123
 
124
  # use raw predictions for mean/quantiles/confidence-interval estimation
 
114
  # For inference_token_len, you can refer to LightGTS (Periodic Patching).
115
  # We recommend to use the period length as the inference_token_len.
116
  output = model.generate(inputs=batch_x,text_input_ids=batch_input_ids,
117
+ text_attention_mask=batch_attention_mask,
118
  text_token_type_ids=batch_token_type_ids,
119
+ max_output_length=forecast_length,
120
+ num_samples=num_samples,
121
+ inference_token_len=48)
122
 
123
 
124
  # use raw predictions for mean/quantiles/confidence-interval estimation