shikhr commited on
Commit
8204068
1 Parent(s): 2dd04ff

allow generation through wrapper model

Browse files
Files changed (1) hide show
  1. 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)