chore: remove generate
Browse files- modeling_hyena.py +0 -4
modeling_hyena.py
CHANGED
@@ -131,10 +131,6 @@ class StripedHyenaModelForCausalLM(StripedHyenaPreTrainedModel):
|
|
131 |
def can_generate(cls) -> bool:
|
132 |
return True
|
133 |
|
134 |
-
def generate(self, input_ids, *args, **kwargs):
|
135 |
-
generations = self.generate(input_ids, *args, **kwargs)
|
136 |
-
strings = self.tokenizer.batch_decode(generations, skip_special_tokens=True)
|
137 |
-
|
138 |
def prepare_inputs_for_generation(
|
139 |
self, input_ids, attention_mask=None, past_key_values=None, **kwargs
|
140 |
):
|
|
|
131 |
def can_generate(cls) -> bool:
|
132 |
return True
|
133 |
|
|
|
|
|
|
|
|
|
134 |
def prepare_inputs_for_generation(
|
135 |
self, input_ids, attention_mask=None, past_key_values=None, **kwargs
|
136 |
):
|