transformers
c74befc9 - HFTracer.trace can now take callables and torch.nn.Module (#18457)

Commit

3 years ago

HFTracer.trace can now take callables and torch.nn.Module (#18457) * Enable HFTracer to trace with custom dummy inputs instead of pre-computed ones * Add HFTracer.trace docstring, and make it possible to handle callable and torch.nn.Module in general * Remove pdb comment * Apply suggestions

References

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

#19449 - [WIP] Fix weights initialization of several vision models

#27720 - Add common processor tests

#29969 - [SigLIP] Add fast tokenizer

#32831 - [Docs] Update resources

#33111 - [Backbone] Remove out_features everywhere

#33174 - [Zero-shot image classification pipeline] Remove tokenizer_kwargs

#39821 - Support MetaCLIP 2

#59 - Fix attention mask handling in EoMT-DINOv3 converter

#41212 - Add EoMT with DINOv3 backbone

#62 - Add initial DEIMv2 model implementation

Author

michaelbenayoun

Parents

fc1d841b

transformers c74befc9 - HFTracer.trace can now take callables and torch.nn.Module (#18457)

transformers
c74befc9 - HFTracer.trace can now take callables and torch.nn.Module (#18457)