[FIX][HuggingfaceLLM]

block.py

@@ -94,4 +94,4 @@ blocks_by_parent_description = swarm.get_by_parent_description(
 
 # Run the block in the swarm
 inference = swarm.run_block(toolagent, "Hello World")
-print(inference)
+print(inference)

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "3.4.1"
+version = "3.4.2"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]

swarms/models/huggingface.py

@@ -131,7 +131,7 @@ class HuggingfaceLLM(AbstractLLM):
         temperature: float = 0.7,
         top_k: int = 40,
         top_p: float = 0.8,
-        dtype = torch.bfloat16,
+        dtype=torch.bfloat16,
         *args,
         **kwargs,
     ):
@@ -189,7 +189,6 @@ class HuggingfaceLLM(AbstractLLM):
                 self.model_id, *args, **kwargs
             ).to(self.device)
 
-
     def print_error(self, error: str):
         """Print error"""
         print(colored(f"Error: {error}", "red"))
@@ -264,7 +263,7 @@ class HuggingfaceLLM(AbstractLLM):
                         *args,
                         **kwargs,
                     )
-                    
+
             return self.tokenizer.decode(
                 outputs[0], skip_special_tokens=True
             )

swarms/structs/blockslist.py

@@ -75,8 +75,8 @@ class BlocksList(BaseStructure):
 
     def get_all(self):
         return self.blocks
-    
-    def run_block(self, block: Any, task: str,  *args, **kwargs):
+
+    def run_block(self, block: Any, task: str, *args, **kwargs):
         """Run the block for the specified task.
 
         Args: