amazon-science / chronos-forecasting

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

Update chronos.py - model.device #11

Closed michaelfeil closed 6 months ago

michaelfeil commented 6 months ago

Issue #, if available: N/A

Description of changes:

Thanks for the very clean impl of the Model, Tokenizer, and Pipeline.

I was curios about it and found a minor improvement in the API - what do you think about it? Feel free to close. Change is untested.

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