removed some extra torch.saves that were being used for debugging

Signed-off-by: kta-intel <[email protected]>

openfl-tutorials/experimental/LLM/neuralchat/Workflow_Interface_NeuralChat.ipynb

@@ -514,10 +514,6 @@
     "        # trainer.train(resume_from_checkpoint=training_args.resume_from_checkpoint)\n",
     "        out = trainer.train()  \n",
     "\n",
-    "        ###############\n",
-    "        torch.save(self.model, 'collaborator_'+self.input+'.pth')\n",
-    "        ###############\n",
-    "        \n",
     "        self.loss = out.training_loss\n",
     "        trainer.save_model()\n",
     "        self.training_completed = True\n",
@@ -548,10 +544,6 @@
     "        print(f'Average local model validation values = {self.local_model_accuracy}')\n",
     "        \n",
     "        self.model = FedAvg([input.model for input in inputs])\n",
-    "\n",
-    "        ################\n",
-    "        torch.save(self.model, 'aggregated_model.pth')\n",
-    "        ################\n",
     "        \n",
     "        self.model.save_pretrained('./aggregated/model')\n",
     "        tokenizer.save_pretrained('./aggregated/tokenizer')\n",