kwargs = cls.fill_params("body", **kwargs)
encoder = kwargs.pop("encoder")
embeddings = kwargs.get("embeddings")
embeddings_common = embeddings.pop("common")
embeddings_order = embeddings.pop("order") or [name for name in kwargs
if "embedding" in name]
decoder = kwargs.pop("decoder")
After Change
for i, embedding in enumerate(embeddings):
embedding_args = {**kwargs, **embedding}
x = cls.embedding(x, name="embedding"+str(i), **embedding_args)
encoder_outputs.append(x)