accelerate
7644a02e - `add_hook_to_module` and `remove_hook_from_module` compatibility with fx.GraphModule (#2369)

Commit

2 years ago

`add_hook_to_module` and `remove_hook_from_module` compatibility with fx.GraphModule (#2369) * fix add & remove hook with torch fx * comment test

References

Author

fxmarty

Parents