fix

psfoley · Sep 29, 2023 · a0c4d1d · a0c4d1d
1 parent 76d4639
commit a0c4d1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/openfl-tutorials/Federated_PyTorch_LLM.ipynb b/openfl-tutorials/Federated_PyTorch_LLM.ipynb
@@ -373,7 +373,8 @@
     "            model_state_dict_key: get_peft_model_state_dict(self.model),\n",
     "            optimizer_state_dict_key: self.optimizer.state_dict()\n",
     "        }\n",
-    "        pt.save(pickle_dict, filepath)"
+    "        pt.save(pickle_dict, filepath)\n",
+    "        "
    ]
   },
   {