riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

added if prediction_length is None

Browse Source
This commit is contained in:
elisim 2023-03-01 19:04:14 +07:00
parent e455bbf5c7
commit b4cbddfa05
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/informer/configuration_informer.py
View File

@ -179,6 +179,9 @@ class InformerConfig(PretrainedConfig):
**kwargs,
):
# time series specific configuration
if prediction_length is None:
raise ValueError("The `prediction_length` should be set.")
self.prediction_length = prediction_length
self.context_length = context_length or prediction_length
self.distribution_output = distribution_output