JonasGeiping commited on
Commit
e8fc157
·
verified ·
1 Parent(s): dce2adb

Update raven_modeling_minimal.py

Browse files
Files changed (1) hide show
  1. raven_modeling_minimal.py +1 -1
raven_modeling_minimal.py CHANGED
@@ -657,7 +657,7 @@ class RavenForCausalLM(RavenPreTrainedModel, GenerationMixin):
657
  print("Dispatching to custom generate function call")
658
  return self.generate_with_adaptive_compute(*args, **kwargs)
659
  else:
660
- super().generate(*args, **kwargs)
661
 
662
  @torch.no_grad()
663
  def generate_minimal(
 
657
  print("Dispatching to custom generate function call")
658
  return self.generate_with_adaptive_compute(*args, **kwargs)
659
  else:
660
+ return super().generate(*args, **kwargs)
661
 
662
  @torch.no_grad()
663
  def generate_minimal(