transformers/tests/models/sam
Yih-Dar ab98f0b0a1
avoid calling gc.collect and cuda.empty_cache (#34514)
* update

* update

* update

* update

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-10-31 16:36:13 +01:00
..
__init__.py Add Segment Anything Model (SAM) (#22654) 2023-04-19 21:01:49 +02:00
test_modeling_sam.py avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00
test_modeling_tf_sam.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_processor_sam.py Remove repeated prepare_images in processor tests (#33163) 2024-09-09 13:20:27 +01:00