[chronos-2]setting group_id by users

[xiaoyu-wang93]

After reading the paper of chronos-2, curious to know whether encourage users to set their own group_id to leverage group attention mechanism?
if yes, Could you provide a more convenient way to pass group_id ? Seems in the code, only _predict_batch() can receive group_id

[shchur]

Group attention is primarily important for efficient training, where we want to operate with a fixed batch size while training the model on groups of different sizes.

At inference time, you can just provide all time series in a single group to predict(), predict_quantiles() or predict_df() and set cross_learning=True. See this comment for an example #429 (comment).

[xiaoyu-wang93]

I see...

for inference, we have a,b,c,d,e 5 time series, a,b,c in a group, d,e in a group in our knowledge, then call predict([a,b,c], cross_learning=True), predict([d,e], cross_learning=True) separately to achieve different group ?

for fine-tuning chronos-2, how can we group time series so that can maximize use the group attention mechanism? is it helpful to improve performance?