Enable memory metrics in tests that need it
PR https://github.com/huggingface/transformers/pull/11851 was merged without updating the tests to reflect the change in the argument default.

Explicitly specified the need for memory metrics for these tests.

Merging now to have CI pass, feel free to comment if that's not the right approach @stas00 @sgugger 
Thanks a lot for catching and fixing @LysandreJik !