|
|
@ -436,7 +436,7 @@
|
|
|
|
"from transformers import AutoModelForCausalLM, AutoTokenizer\n",
|
|
|
|
"from transformers import AutoModelForCausalLM, AutoTokenizer\n",
|
|
|
|
"\n",
|
|
|
|
"\n",
|
|
|
|
"from swarms import ToolAgent\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",
|
|
|
|
"\n",
|
|
|
|
"# Load the pre-trained model and tokenizer\n",
|
|
|
|
"# Load the pre-trained model and tokenizer\n",
|
|
|
|
"model = AutoModelForCausalLM.from_pretrained(\n",
|
|
|
|
"model = AutoModelForCausalLM.from_pretrained(\n",
|
|
|
|