diff --git a/swarms/__init__.py b/swarms/__init__.py index ae28e59e..65ae8c28 100644 --- a/swarms/__init__.py +++ b/swarms/__init__.py @@ -4,7 +4,6 @@ from swarms.orchestrator.autoscaler import AutoScaler # worker # from swarms.workers.worker_node import WorkerNode from swarms.workers.worker import Worker -from swarms.workers.autobot import AutoBot #boss from swarms.boss.boss_node import BossNode diff --git a/swarms/models/huggingface.py b/swarms/models/huggingface.py index f4681a8e..fe88ea58 100644 --- a/swarms/models/huggingface.py +++ b/swarms/models/huggingface.py @@ -5,7 +5,13 @@ import logging from transformers import AutoModelForCausalLM, AutoTokenizer class HuggingFaceLLM: - def __init__(self, model_id: str, device: str = None, max_length: int = 20, quantize: bool = False, quantization_config: dict = None): + def __init__(self, + model_id: str, + device: str = None, + max_length: int = 20, + quantize: bool = False, + quantization_config: dict = None + ): self.logger = logging.getLogger(__name__) self.device = device if device else ('cuda' if torch.cuda.is_available() else 'cpu') self.model_id = model_id