Upload ByteETM-Korean (HF inference compatible)
Browse files- 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)
|