Zymrael commited on
Commit
86e226f
1 Parent(s): cb37c8c

chore: remove generate

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