amazon-science / chronos-forecasting

Chronos: Pretrained (Language) Models for Probabilistic Time Series Forecasting
https://arxiv.org/abs/2403.07815
Apache License 2.0
2.02k stars 238 forks source link

Relax torch and transformers versions #81

Closed abdulfatir closed 1 month ago

abdulfatir commented 1 month ago

Description of changes: This PR relaxes torch and transformers versions to allow for older versions that were used during original training. This is needed in light of recent torch/transformers versions being slower with DDP.

Relevant issues (but the problem may be deeper than these):

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

lostella commented 1 month ago

@abdulfatir could you just link in the OP all the issues related to performance regressions?

abdulfatir commented 1 month ago

@abdulfatir could you just link in the OP all the issues related to performance regressions?

Done.