Update raven_modeling_minimal.py
Browse files
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(
|