Upload model

modeling_genbio.py

@@ -1,4 +1,3 @@
-import warnings
 from lightning.pytorch import LightningModule
 from lightning.pytorch.core.saving import _load_state
 from transformers import PreTrainedModel, PretrainedConfig
@@ -32,9 +31,7 @@ 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.
-        with warnings.catch_warnings():
-            warnings.simplefilter("ignore")
-            self.genbio_model = _load_state(genbio_cls, checkpoint, strict_loading=False)
+        self.genbio_model = _load_state(genbio_cls, checkpoint, strict_loading=False)
 
     @classmethod
     def from_genbio_model(cls, model: LightningModule):