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

[WIP] Move X-MOD models to facebook organization (#21640)

Browse Source 
Move X-MOD models to facebook org
This commit is contained in:
Jannis Vamvas
2023-02-16 15:18:25 +01:00
committed by GitHub
parent 751f17aa48
commit 61abe3290b
4 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/xmod/test_modeling_xmod.py
View File

@@ -512,7 +512,7 @@ class XmodModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
class XmodModelIntegrationTest(unittest.TestCase):
@slow
def test_xmod_base(self):
model = XmodModel.from_pretrained("jvamvas/xmod-base")
model = XmodModel.from_pretrained("facebook/xmod-base")
# language en_XX
model.set_default_language("en_XX")
@@ -545,7 +545,7 @@ class XmodModelIntegrationTest(unittest.TestCase):
@slow
def test_xmod_large_prenorm(self):
model = XmodModel.from_pretrained("jvamvas/xmod-large-prenorm")
model = XmodModel.from_pretrained("facebook/xmod-large-prenorm")
# language en_XX
model.set_default_language("en_XX")
@@ -581,7 +581,7 @@ class XmodModelIntegrationTest(unittest.TestCase):
@slow
def test_multilingual_batch(self):
model = XmodModel.from_pretrained("jvamvas/xmod-base")
model = XmodModel.from_pretrained("facebook/xmod-base")
# fmt: off
input_ids = torch.tensor([
[0, 581, 10269, 83, 99942, 136, 60742, 23, 70, 80583, 18276, 2],
@@ -608,7 +608,7 @@ class XmodModelIntegrationTest(unittest.TestCase):
@slow
def test_end_to_end_mask_fill(self):
tokenizer = XLMRobertaTokenizer.from_pretrained("xlm-roberta-base")
model = XmodForMaskedLM.from_pretrained("jvamvas/xmod-base", default_language="en_XX")
model = XmodForMaskedLM.from_pretrained("facebook/xmod-base", default_language="en_XX")
model.to(torch_device)
sentences = [