transformers/tests/models/sam
Hz, Ji 50378cbf6c
device agnostic models testing (#27146)
* device agnostic models testing

* add decorator `require_torch_fp16`

* make style

* apply review suggestion

* Oops, the fp16 decorator was misused
2023-10-31 18:12:14 +01:00
..
__init__.py Add Segment Anything Model (SAM) (#22654) 2023-04-19 21:01:49 +02:00
test_modeling_sam.py device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
test_modeling_tf_sam.py Stop storing references to bound methods via tf.function (#24146) 2023-06-13 19:04:22 +01:00
test_processor_sam.py TF port of the Segment Anything Model (SAM) (#22970) 2023-05-19 14:14:13 +01:00