import tensorflow as tf
from tensorflow.keras.optimizers import Adam

# Create and compile the model
model = create_transformer_model(sequence_length=10, d_model=32, num_heads=2, ff_dim=64, num_layers=1)
model.compile(optimizer=Adam(learning_rate=0.001), loss='mse')

# Train the model
model.fit(dataset, epochs=9)