src/transformers/utils/fx.py/HFTracer/call_module
class HFTracer:    def call_module(self, m, forward, args, kwargs):
        self.orig_forward = forward
        return super().call_module(m, forward, args, kwargs)