Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321)
* Rename to test_model_common_attributes The method name is misleading - it is testing being able to get and set embeddings, not common attributes to all models * Explicitly skip
This commit is contained in:
@@ -1971,13 +1971,17 @@ class ModelTesterMixin:
|
||||
# Check that the model can still do a forward pass successfully (every parameter should be resized)
|
||||
model(**self._prepare_for_class(inputs_dict, model_class))
|
||||
|
||||
def test_model_common_attributes(self):
|
||||
def test_model_get_set_embeddings(self):
|
||||
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
|
||||
|
||||
for model_class in self.all_model_classes:
|
||||
model = model_class(config)
|
||||
self.assertIsInstance(model.get_input_embeddings(), (nn.Embedding, AdaptiveEmbedding))
|
||||
model.set_input_embeddings(nn.Embedding(10, 10))
|
||||
|
||||
new_input_embedding_layer = nn.Embedding(10, 10)
|
||||
model.set_input_embeddings(new_input_embedding_layer)
|
||||
self.assertEqual(model.get_input_embeddings(), new_input_embedding_layer)
|
||||
|
||||
x = model.get_output_embeddings()
|
||||
self.assertTrue(x is None or isinstance(x, nn.Linear))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user