Update estimator.py to enable cpu support

time-series-foundation-models / lag-llama

Lag-Llama: Towards Foundation Models for Probabilistic Time Series Forecasting

Apache License 2.0

1.09k stars 121 forks source link

Closed AmosG closed 2 months ago

kashif commented 4 months ago

thanks will review soon!

ashok-arjun commented 4 months ago

Thanks @AmosG for the PR.

Ideally we want to have a device parameter that we can set (so one can use CPU despite having a GPU etc.).

ashok-arjun commented 2 months ago