fix inference_app.py

inference_app.py

@@ -104,6 +104,7 @@ def get_system(system_id: str) -> PinderSystem:
     return PinderSystem(system_id)
 from Bio import PDB
 from Bio.PDB.PDBIO import PDBIO
+from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
 
 def extract_coordinates_from_pdb(filename):
     r"""
@@ -873,8 +874,14 @@ class FinalMPNNModelight(pl.LightningModule):
             },
         }
 
-model_path = "/home/sukanya/iitm_bisect_pinder_submission/EquiMPNN-epoch=413-val_loss=9.25-val_acc=0.00.ckpt"
-model = FinalMPNNModelight.load_from_checkpoint(model_path)
+# model_path = "/home/sukanya/iitm_bisect_pinder_submission/EquiMPNN-epoch=413-val_loss=9.25-val_acc=0.00.ckpt"
+# model = FinalMPNNModelight.load_from_checkpoint(model_path)
+model = FinalMPNNModelight()
+model = FSDP(model)
+checkpoint = torch.load("/home/sukanya/iitm_bisect_pinder_submission/EquiMPNN-epoch=413-val_loss=9.25-val_acc=0.00.ckpt")
+model_state_dict = checkpoint['state_dict'] 
+model.load_state_dict(model_state_dict)
+
 trainer = pl.Trainer(