Fix PyTorch RAG tests GPU OOM (#16881)

Commit

3 years ago

Fix PyTorch RAG tests GPU OOM (#16881) * add torch.cuda.empty_cache in some PT RAG tests * torch.cuda.empty_cache in tearDownModule() * tearDown() * add gc.collect() Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

References

#16881 - Fix PyTorch RAG tests GPU OOM

#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

#58 - Add EoMT DINOv3 model

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

#41212 - Add EoMT with DINOv3 backbone

#62 - Add initial DEIMv2 model implementation

Author

ydshieh

Parents

3e47d19c

transformers 32adbb26 - Fix PyTorch RAG tests GPU OOM (#16881)

transformers
32adbb26 - Fix PyTorch RAG tests GPU OOM (#16881)