jadechoghari
/

mar

jadechoghari commited on Sep 8

Commit

303a347

•

1 Parent(s): 61da904

Update modeling.py

Files changed (1) hide show

modeling.py CHANGED Viewed

@@ -166,7 +166,7 @@ class MARModel(PreTrainedModel):
         model_architecture.load_state_dict(state_dict, strict=False)
         # update this so the model works on the forward call
-        self.model = model_architecture
         model.eval()
         return model

         model_architecture.load_state_dict(state_dict, strict=False)
         # update this so the model works on the forward call
+        model = model_architecture
         model.eval()
         return model