From 772460f5e3aaad478700940e985ac455c0437971 Mon Sep 17 00:00:00 2001 From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com> Date: Sun, 18 May 2025 06:48:42 +0000 Subject: [PATCH] fix: remove transformers import and tokenizer restoration logic from agent_non_serializable.py (resolves undefined import error) --- swarms/structs/agent_non_serializable.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/swarms/structs/agent_non_serializable.py b/swarms/structs/agent_non_serializable.py index e494d144..f3bdec9e 100644 --- a/swarms/structs/agent_non_serializable.py +++ b/swarms/structs/agent_non_serializable.py @@ -9,7 +9,6 @@ Usage: restore_non_serializable_properties(agent) """ -from transformers import AutoTokenizer from concurrent.futures import ThreadPoolExecutor import logging @@ -34,12 +33,15 @@ def restore_non_serializable_properties(agent): This should be called after loading agent state from disk. """ # Restore tokenizer if model_name is available - if getattr(agent, "model_name", None): - try: - agent.tokenizer = AutoTokenizer.from_pretrained(agent.model_name) - except Exception: - agent.tokenizer = None - else: + agent.tokenizer = None + try: + if getattr(agent, "model_name", None): + try: + from transformers import AutoTokenizer + agent.tokenizer = AutoTokenizer.from_pretrained(agent.model_name) + except Exception: + agent.tokenizer = None + except ImportError: agent.tokenizer = None # Restore long_term_memory (dummy for demo, replace with real backend as needed)