[NLLB-MoE] model_type update for auto mapping (#22470)

edit default model type and testing path set to hf-internal-testing
This commit is contained in:
Arthur
2023-03-30 15:36:07 +02:00
committed by GitHub
parent 11426641dc
commit 349e1242d9
2 changed files with 4 additions and 4 deletions

View File

@@ -354,14 +354,14 @@ class NllbMoeModelIntegrationTests(unittest.TestCase):
@cached_property
def tokenizer(self):
return NllbTokenizer.from_pretrained("ArthurZ/random-nllb-moe-2-experts")
return NllbTokenizer.from_pretrained("hf-internal-testing/random-nllb-moe-2-experts")
@cached_property
def big_model(self):
return NllbMoeForConditionalGeneration.from_pretrained("facebook/nllb-moe-54b")
def inference_no_head(self):
model = NllbMoeModel.from_pretrained("ArthurZ/random-nllb-moe-2-experts").eval()
model = NllbMoeModel.from_pretrained("hf-internal-testing/random-nllb-moe-2-experts").eval()
with torch.no_grad():
output = model(**self.model_inputs)
# fmt: off
@@ -382,7 +382,7 @@ class NllbMoeModelIntegrationTests(unittest.TestCase):
and `transformers` implementation of NLLB-MoE transformers. We only check the logits
of the second sample of the batch, as it is padded.
"""
model = NllbMoeForConditionalGeneration.from_pretrained("ArthurZ/random-nllb-moe-2-experts").eval()
model = NllbMoeForConditionalGeneration.from_pretrained("hf-internal-testing/random-nllb-moe-2-experts").eval()
with torch.no_grad():
output = model(**self.model_inputs)