allow generation through wrapper model
Browse files- mgpt_modelling.py +3 -0
mgpt_modelling.py
CHANGED
@@ -12,3 +12,6 @@ class MusicModel(PreTrainedModel):
|
|
12 |
|
13 |
def forward(self, inputs):
|
14 |
return self.model(inputs)
|
|
|
|
|
|
|
|
12 |
|
13 |
def forward(self, inputs):
|
14 |
return self.model(inputs)
|
15 |
+
|
16 |
+
def generate(self, inputs):
|
17 |
+
return self.model.generate(inputs)
|