src/transformers/benchmark/benchmark_utils.py/Benchmark/inference_memory
class Benchmark:    def inference_memory(self, *args, **kwargs) -> [Memory, Optional[MemorySummary]]:
        return separate_process_wrapper_fn(self._inference_memory, self.args.do_multi_processing)(*args, **kwargs)