Batch processing

[agemagician]

Hello,

Thanks for your great work.

I have noticed that you don't use batch processing in your "get_cell_types_for_adata" function, which makes the feature extraction process very slow.

Compared to the scGPT, which processes inputs in batches, it is 24 times faster.

Do you have any plans to support batch processing ?