Typo "UNKWOWN" -> "UNKNOWN" (#13675)

This commit is contained in:
Kamal Raj
2021-09-21 18:41:26 +05:30
committed by GitHub
parent 78807d86eb
commit 8d533e6ad6
7 changed files with 14 additions and 14 deletions

View File

@@ -20,7 +20,7 @@ import unittest
from transformers import is_torch_available
from transformers.testing_utils import (
DUMMY_UNKWOWN_IDENTIFIER,
DUMMY_UNKNOWN_IDENTIFIER,
SMALL_MODEL_IDENTIFIER,
require_scatter,
require_torch,
@@ -261,7 +261,7 @@ class AutoModelTest(unittest.TestCase):
self.assertEqual(model.num_parameters(only_trainable=True), 14410)
def test_from_identifier_from_model_type(self):
model = AutoModelWithLMHead.from_pretrained(DUMMY_UNKWOWN_IDENTIFIER)
model = AutoModelWithLMHead.from_pretrained(DUMMY_UNKNOWN_IDENTIFIER)
self.assertIsInstance(model, RobertaForMaskedLM)
self.assertEqual(model.num_parameters(), 14410)
self.assertEqual(model.num_parameters(only_trainable=True), 14410)