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

fix: ckpt paths. (#19159)

Browse Source
This commit is contained in:
Sayak Paul
2022-09-22 20:33:01 +05:30
committed by GitHub
parent 74a3ea4737
commit 3a396c59b8
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/vit_msn/modeling_vit_msn.py
View File

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

4
tests/models/vit_msn/test_modeling_vit_msn.py
View File

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