Update Mo.py

Mo.py

@@ -196,7 +196,7 @@ class Block(layers.Layer):
         x = self.lo(x)
         return x
 
-class ReLM(tf.keras.Model):
+class C(tf.keras.Model):
     def __init__(self, vocab_size, max_seq_len, d_model, n_layers, dropout_rate=0.1):
         super().__init__()
         self.token_embedding = layers.Embedding(vocab_size, d_model)