comment some codes in transformer2 for the tests