def get_intermediate_layers(self, x, n=1):
    x = self.prepare_tokens(x)
    output = []
    for (i, blk) in enumerate(self.blocks):
        x = blk(x)
        if ((len(self.blocks) - i) <= n):
            output.append(self.norm(x))
    return output
def get_intermediate_layers(self, x, n=1):
    x = self.prepare_tokens(x)
    output = []
    for (i, blk) in enumerate(self.blocks):
        x = blk(x)
        if ((len(self.blocks) - i) <= n):
            output.append(self.norm(x))
    return output