fix input bug

mgpt_modelling.py

@@ -10,8 +10,8 @@ class MusicModel(PreTrainedModel):
         super().__init__(config)
         self.model = GPT(config)
 
-    def forward(self, inputs):
-        return self.model(inputs)
+    def forward(self, *args, **kwargs):
+        return self.model(*args, **kwargs)
 
-    def generate(self, inputs):
-        return self.model.generate(inputs)
+    def generate(self, *args, **kwargs):
+        return self.model.generate(*args, **kwargs)