Skip to content

GitLab

Unverified Commit 74b846ca authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Update `(TF)SamModelIntegrationTest` (#24199) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent d7389cd2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/models/sam/test_modeling_tf_sam.py
View file @ 74b846ca
...@@ -428,8 +428,9 @@ def prepare_dog_img(): ...@@ -428,8 +428,9 @@ def prepare_dog_img():
return raw_image return raw_image
@require_tf
@slow @slow
class SamModelIntegrationTest(unittest.TestCase): class TFSamModelIntegrationTest(unittest.TestCase):
def tearDown(self): def tearDown(self):
super().tearDown() super().tearDown()
# clean-up as much as possible GPU memory occupied by PyTorch # clean-up as much as possible GPU memory occupied by PyTorch
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment