Update bucket_memory_model.py

bucket_memory_model.py

@@ -20,6 +20,7 @@ class BucketMemoryConfig(PretrainedConfig):
     ):
         super().__init__(**kwargs)
         self.vocab_size = vocab_size
+        self.hidden_size = hidden_size
         self.d_model = d_model
         self.num_layers = num_layers
         self.num_buckets = num_buckets