diff --git a/playground/collabs/swarms_example.ipynb b/playground/collabs/swarms_example.ipynb index c0f52ed1..c29440c7 100644 --- a/playground/collabs/swarms_example.ipynb +++ b/playground/collabs/swarms_example.ipynb @@ -436,7 +436,7 @@ "from transformers import AutoModelForCausalLM, AutoTokenizer\n", "\n", "from swarms import ToolAgent\n", - "from swarms.utils.json_utils import base_model_to_json\n", + "from swarms.utils.jsonl_utils import base_model_to_json\n", "\n", "# Load the pre-trained model and tokenizer\n", "model = AutoModelForCausalLM.from_pretrained(\n",