fix input bug
Browse files- mgpt_modelling.py +4 -4
mgpt_modelling.py
CHANGED
@@ -10,8 +10,8 @@ class MusicModel(PreTrainedModel):
|
|
10 |
super().__init__(config)
|
11 |
self.model = GPT(config)
|
12 |
|
13 |
-
def forward(self,
|
14 |
-
return self.model(
|
15 |
|
16 |
-
def generate(self,
|
17 |
-
return self.model.generate(
|
|
|
10 |
super().__init__(config)
|
11 |
self.model = GPT(config)
|
12 |
|
13 |
+
def forward(self, *args, **kwargs):
|
14 |
+
return self.model(*args, **kwargs)
|
15 |
|
16 |
+
def generate(self, *args, **kwargs):
|
17 |
+
return self.model.generate(*args, **kwargs)
|