Fix 29807 sinusoidal positional encodings in Flaubert, Informer and XLM (#29904)
* Fix sinusoidal_embeddings in FlaubertModel * Fix for Informer * Fix for XLM * Move sinusoidal emb for XLM * Move sinusoidal emb for Flaubert * Small cleanup * Add comments on tests code copied from * Add with Distilbert->
This commit is contained in:
committed by
GitHub
parent
83b26dd79d
commit
416711c3ea
@@ -35,7 +35,11 @@ if is_torch_available():
|
||||
import torch
|
||||
|
||||
from transformers import InformerConfig, InformerForPrediction, InformerModel
|
||||
from transformers.models.informer.modeling_informer import InformerDecoder, InformerEncoder
|
||||
from transformers.models.informer.modeling_informer import (
|
||||
InformerDecoder,
|
||||
InformerEncoder,
|
||||
InformerSinusoidalPositionalEmbedding,
|
||||
)
|
||||
|
||||
|
||||
@require_torch
|
||||
@@ -164,6 +168,12 @@ class InformerModelTester:
|
||||
|
||||
self.parent.assertTrue((encoder_last_hidden_state_2 - encoder_last_hidden_state).abs().max().item() < 1e-3)
|
||||
|
||||
embed_positions = InformerSinusoidalPositionalEmbedding(
|
||||
config.context_length + config.prediction_length, config.d_model
|
||||
)
|
||||
self.parent.assertTrue(torch.equal(model.encoder.embed_positions.weight, embed_positions.weight))
|
||||
self.parent.assertTrue(torch.equal(model.decoder.embed_positions.weight, embed_positions.weight))
|
||||
|
||||
with tempfile.TemporaryDirectory() as tmpdirname:
|
||||
decoder = model.get_decoder()
|
||||
decoder.save_pretrained(tmpdirname)
|
||||
|
||||
Reference in New Issue
Block a user