SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update (TF)SamModelIntegrationTest (#24199)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-06-13 14:28:14 +02:00
committed by GitHub
parent d7389cd201
commit 74b846cacf
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/sam/test_modeling_tf_sam.py
View File

@@ -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