transformers
HFTracer.trace can now take callables and torch.nn.Module
#18457

Merged

HFTracer.trace can now take callables and torch.nn.Module #18457

michaelbenayoun merged 4 commits into huggingface:main from michaelbenayoun:fx_hf_tracer_dummy_inputs

Enable HFTracer to trace with custom dummy inputs instead of pre-comp…

5ff7eaf1

Add HFTracer.trace docstring, and make it possible to handle callable…

3d1ba72a

Remove pdb comment

70342c7a

michaelbenayoun requested a review from

sgugger 4 years ago

thomasw21 commented on 2022-08-03

sgugger approved these changes on 2022-08-03

Apply suggestions

40822116

michaelbenayoun merged c74befc9 into main 4 years ago

michaelbenayoun deleted the fx_hf_tracer_dummy_inputs branch 4 years ago

Reviewers

sgugger

thomasw21

Assignees

No one assigned

Labels

None yet

Milestone

No milestone