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
3a396c59
Unverified
Commit
3a396c59
authored
Sep 22, 2022
by
Sayak Paul
Committed by
GitHub
Sep 22, 2022
Browse files
fix: ckpt paths. (#19159)
parent
74a3ea47
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
src/transformers/models/vit_msn/modeling_vit_msn.py
src/transformers/models/vit_msn/modeling_vit_msn.py
+2
-2
tests/models/vit_msn/test_modeling_vit_msn.py
tests/models/vit_msn/test_modeling_vit_msn.py
+2
-2
No files found.
src/transformers/models/vit_msn/modeling_vit_msn.py
View file @
3a396c59
...
@@ -36,9 +36,9 @@ logger = logging.get_logger(__name__)
...
@@ -36,9 +36,9 @@ logger = logging.get_logger(__name__)
_CONFIG_FOR_DOC
=
"ViTMSNConfig"
_CONFIG_FOR_DOC
=
"ViTMSNConfig"
_CHECKPOINT_FOR_DOC
=
"
sayakpaul
/vit-msn-small"
_CHECKPOINT_FOR_DOC
=
"
facebook
/vit-msn-small"
VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
"
sayakpaul
/vit-msn-small"
,
"
facebook
/vit-msn-small"
,
# See all ViTMSN models at https://huggingface.co/models?filter=vit_msn
# See all ViTMSN models at https://huggingface.co/models?filter=vit_msn
]
]
...
...
tests/models/vit_msn/test_modeling_vit_msn.py
View file @
3a396c59
...
@@ -215,12 +215,12 @@ def prepare_img():
...
@@ -215,12 +215,12 @@ def prepare_img():
class
ViTMSNModelIntegrationTest
(
unittest
.
TestCase
):
class
ViTMSNModelIntegrationTest
(
unittest
.
TestCase
):
@
cached_property
@
cached_property
def
default_feature_extractor
(
self
):
def
default_feature_extractor
(
self
):
return
ViTFeatureExtractor
.
from_pretrained
(
"
sayakpaul
/vit-msn-small"
)
if
is_vision_available
()
else
None
return
ViTFeatureExtractor
.
from_pretrained
(
"
facebook
/vit-msn-small"
)
if
is_vision_available
()
else
None
@
slow
@
slow
def
test_inference_image_classification_head
(
self
):
def
test_inference_image_classification_head
(
self
):
torch
.
manual_seed
(
2
)
torch
.
manual_seed
(
2
)
model
=
ViTMSNForImageClassification
.
from_pretrained
(
"
sayakpaul
/vit-msn-small"
).
to
(
torch_device
)
model
=
ViTMSNForImageClassification
.
from_pretrained
(
"
facebook
/vit-msn-small"
).
to
(
torch_device
)
feature_extractor
=
self
.
default_feature_extractor
feature_extractor
=
self
.
default_feature_extractor
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