diff --git a/swarms/artifacts/error_artifact.py b/swarms/artifacts/error_artifact.py index fc5cbcb7..01cf1b3e 100644 --- a/swarms/artifacts/error_artifact.py +++ b/swarms/artifacts/error_artifact.py @@ -1,16 +1,16 @@ -#from shapeless import shapeless +# #from shapeless import shapeless -#@shapeless -class ErrorArtifact: - def __init__( - self, - value - ): - self.value = value +# #@shapeless +# class ErrorArtifact: +# def __init__( +# self, +# value +# ): +# self.value = value - def __add__(self, other): - return ErrorArtififact(self.value + other.value) +# def __add__(self, other): +# return ErrorArtififact(self.value + other.value) - def to_text(self) -> str: - return self.value +# def to_text(self) -> str: +# return self.value \ No newline at end of file diff --git a/swarms/models/petals.py b/swarms/models/petals.py index 56b10ef1..55d38eaa 100644 --- a/swarms/models/petals.py +++ b/swarms/models/petals.py @@ -3,7 +3,16 @@ from transformers import AutoTokenizer, AutoModelForCausalLM class Petals: """Petals Bloom models.""" - def __init__(self, model_name="bigscience/bloom-petals", temperature=0.7, max_new_tokens=256, top_p=0.9, top_k=None, do_sample=True, max_length=None): + def __init__( + self, + model_name="bigscience/bloom-petals", + temperature=0.7, + max_new_tokens=256, + top_p=0.9, + top_k=None, + do_sample=True, + max_length=None + ): self.model_name = model_name self.temperature = temperature self.max_new_tokens = max_new_tokens diff --git a/swarms/workflows/main.py b/swarms/workflows/main.py index 61f224c6..4bcc7ff1 100644 --- a/swarms/workflows/main.py +++ b/swarms/workflows/main.py @@ -9,7 +9,6 @@ from logging import Logger from typing import Optional, Union from rich.logging import RichHandler -#from shapeless import shapeless from swarms.artifacts.error_artifact import ErrorArtifact from swarms.structs.task import BaseTask