Modify network architecture from default model

I am using the HF Decision Transformer implementation, is there a way to remove one of the 3
DecisionTransformerGPT2Block shown in the figure from this model? And is it something that could be useful? My idea was that I had a too expressive/complex architecture for my problem.