Upload model

README.md

@@ -1,3 +1,6 @@
+---
+{}
+---
 # Load model for inference
 ```python
 import torch

modeling_genbio.py

@@ -1,3 +1,5 @@
+import warnings
+
 from lightning.pytorch import LightningModule
 from lightning.pytorch.core.saving import _load_state
 from transformers import PreTrainedModel, PretrainedConfig
@@ -31,7 +33,9 @@ class GenBioModel(PreTrainedModel):
         # TODO: _load_state is a private function and it throws a warning for an
         # empty state_dict. We need a fucntion to intialize our model; this
         # is the only choice we have for now.
-        self.genbio_model = _load_state(genbio_cls, checkpoint, strict_loading=False)
+        with warnings.catch_warnings():
+            warnings.filterwarnings("ignore", "Found keys that are*")
+            self.genbio_model = _load_state(genbio_cls, checkpoint, strict_loading=False)
 
     @classmethod
     def from_genbio_model(cls, model: LightningModule):