From cb5b103253e7e28b0b1acff5fb47c8820d6223d5 Mon Sep 17 00:00:00 2001 From: Kye Date: Sun, 30 Jul 2023 08:59:36 -0400 Subject: [PATCH] clean up --- swarms/agents/models/huggingface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swarms/agents/models/huggingface.py b/swarms/agents/models/huggingface.py index cd73d665..a69a4efd 100644 --- a/swarms/agents/models/huggingface.py +++ b/swarms/agents/models/huggingface.py @@ -27,7 +27,7 @@ class HuggingFaceLLM: except Exception as e: self.logger.error(f"Failed to load the model or the tokenizer: {e}") raise - def generate_text(self, prompt_text: str, max_length: int = None): + def generate(self, prompt_text: str, max_length: int = None): max_length = max_length if max_length else self.max_length try: inputs = self.tokenizer.encode(prompt_text, return_tensors="pt").to(self.device)