idah4 commited on
Commit
9b241bc
·
verified ·
1 Parent(s): 8d8a77c

Upload ByteETM-Korean (HF inference compatible)

Browse files
Files changed (1) hide show
  1. modeling_byteetm.py +3 -0
modeling_byteetm.py CHANGED
@@ -196,3 +196,6 @@ class HFByteETM(PreTrainedModel):
196
  def forward(self, input_ids, **kwargs):
197
  logits, _ = self.model(input_ids)
198
  return {"logits": logits}
 
 
 
 
196
  def forward(self, input_ids, **kwargs):
197
  logits, _ = self.model(input_ids)
198
  return {"logits": logits}
199
+
200
+ def generate(self, *args, **kwargs): # <── 추가
201
+ return self.model.generate(*args, **kwargs)