Num_query_channels low compared to original implementation

Hey, by browsing the code I found that the num_query_chanels for the output is quite low (293), for the original perceiver IO, it is around 768 on text data and even higher for optical flow applications, isn't there a way we could increase those ?