Update raven_modeling_minimal.py

raven_modeling_minimal.py

@@ -19,7 +19,7 @@ import torch.nn.functional as F
 from transformers import GenerationConfig
 
 
-class RavenPreTrainedModel(PreTrainedModel):
+class RavenPreTrainedModel(PreTrainedModel, GenerationMixin):
     config_class = RavenConfig
     base_model_prefix = "model"
     supports_gradient_checkpointing = True
@@ -32,7 +32,8 @@ class RavenPreTrainedModel(PreTrainedModel):
     _supports_static_cache = False
 
     def _init_weights(self, module):
-        print("Random Initialization not implemented.")
+        if not torch.rand((1,)).is_meta:
+            print("Random Initialization not implemented.")
 
 
 @dataclass