Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
2e9af294
Unverified
Commit
2e9af294
authored
Jan 05, 2022
by
Suraj Patil
Committed by
GitHub
Jan 05, 2022
Browse files
[CLIP] Fix TF test (#15042)
parent
443fdaf2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
4 deletions
+5
-4
tests/test_modeling_tf_clip.py
tests/test_modeling_tf_clip.py
+5
-4
No files found.
tests/test_modeling_tf_clip.py
View file @
2e9af294
...
@@ -254,7 +254,7 @@ class TFCLIPVisionModelTest(TFModelTesterMixin, unittest.TestCase):
...
@@ -254,7 +254,7 @@ class TFCLIPVisionModelTest(TFModelTesterMixin, unittest.TestCase):
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
for
model_name
in
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST
[:
1
]:
for
model_name
in
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST
[:
1
]:
model
=
TFCLIPVisionModel
.
from_pretrained
(
model_name
,
from_pt
=
True
)
model
=
TFCLIPVisionModel
.
from_pretrained
(
model_name
)
self
.
assertIsNotNone
(
model
)
self
.
assertIsNotNone
(
model
)
...
@@ -359,7 +359,7 @@ class TFCLIPTextModelTest(TFModelTesterMixin, unittest.TestCase):
...
@@ -359,7 +359,7 @@ class TFCLIPTextModelTest(TFModelTesterMixin, unittest.TestCase):
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
for
model_name
in
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST
[:
1
]:
for
model_name
in
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST
[:
1
]:
model
=
TFCLIPTextModel
.
from_pretrained
(
model_name
,
from_pt
=
True
)
model
=
TFCLIPTextModel
.
from_pretrained
(
model_name
)
self
.
assertIsNotNone
(
model
)
self
.
assertIsNotNone
(
model
)
...
@@ -618,7 +618,7 @@ class TFCLIPModelTest(TFModelTesterMixin, unittest.TestCase):
...
@@ -618,7 +618,7 @@ class TFCLIPModelTest(TFModelTesterMixin, unittest.TestCase):
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
for
model_name
in
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST
[:
1
]:
for
model_name
in
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST
[:
1
]:
model
=
TFCLIPModel
.
from_pretrained
(
model_name
,
from_pt
=
True
)
model
=
TFCLIPModel
.
from_pretrained
(
model_name
)
self
.
assertIsNotNone
(
model
)
self
.
assertIsNotNone
(
model
)
...
@@ -630,11 +630,12 @@ def prepare_img():
...
@@ -630,11 +630,12 @@ def prepare_img():
@
require_vision
@
require_vision
@
require_tf
class
TFCLIPModelIntegrationTest
(
unittest
.
TestCase
):
class
TFCLIPModelIntegrationTest
(
unittest
.
TestCase
):
@
slow
@
slow
def
test_inference
(
self
):
def
test_inference
(
self
):
model_name
=
"openai/clip-vit-base-patch32"
model_name
=
"openai/clip-vit-base-patch32"
model
=
TFCLIPModel
.
from_pretrained
(
model_name
,
from_pt
=
True
)
model
=
TFCLIPModel
.
from_pretrained
(
model_name
)
processor
=
CLIPProcessor
.
from_pretrained
(
model_name
)
processor
=
CLIPProcessor
.
from_pretrained
(
model_name
)
image
=
prepare_img
()
image
=
prepare_img
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment