The avaliable names for the "output_layer" in forward

[drcoke123]

Hi, I am using the NIM endpoint checking for evo2, and I am using run model forward pass.

From the document, I only know that "embedding_layer" and "unembed" are the available options for the output layer. As the main target for me is to get the logit value from the model with a DNA seq provided, may I ask what the other available layer names or options are for us to request?