From e9f2b6e9135097428e7024f9738310365a6e7af0 Mon Sep 17 00:00:00 2001 From: Kye Date: Sun, 30 Jul 2023 08:59:36 -0400 Subject: [PATCH] clean up Former-commit-id: 5cf0e201f803a1308f1b1b7ef20065b9615d05b0 --- 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)