You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
It is not clear from the documentation and the sample code, if the forecast generation can be performed on a GPU, multiple GPUs, or multiple GPUs in multiple nodes. If this is the case, please add some descriptions on how to achieve this.
Describe the solution you'd like
Maybe a simple device_map="auto" argument support in the from_pretrained method that automatically loads the model and the subsequent data onto a GPU/s.
Additional context
I respect the great work you have done! Especially around enabling multivariate modelling.
The text was updated successfully, but these errors were encountered:
ryuta-yoshimatsu
changed the title
GPU, single-node multi-GPU, multi-node multi-GPU support
GPU, single-node multi-GPU, multi-node multi-GPU support for forecast generation
May 31, 2024
Is your feature request related to a problem? Please describe.
It is not clear from the documentation and the sample code, if the forecast generation can be performed on a GPU, multiple GPUs, or multiple GPUs in multiple nodes. If this is the case, please add some descriptions on how to achieve this.
Describe the solution you'd like
Maybe a simple device_map="auto" argument support in the from_pretrained method that automatically loads the model and the subsequent data onto a GPU/s.
Additional context
I respect the great work you have done! Especially around enabling multivariate modelling.
The text was updated successfully, but these errors were encountered: