diff --git a/pyai.py b/pyai.py index 827a9b3..f48ce57 100644 --- a/pyai.py +++ b/pyai.py @@ -62,7 +62,7 @@ def Softmax(self, x: list | Tensor): soft = nn.Softmax(dim=1).to(self.GPU)(x) return soft - def Sigmod(self, x: list | Tensor): + def Sigmoid(self, x: list | Tensor): if isinstance(x, list): tensor = Tensor(x, 1).to(self.GPU) sigmod = nn.Sigmoid().to(self.GPU)(tensor) @@ -168,4 +168,4 @@ def BERT(self, text: str, model: str = "bert-base-uncased"): class Transformers: def __init__(self, text: str, typeOfOperation: str = "text-generation", model: str = "gpt2"): transformer = pipeline(typeOfOperation, model=model) - return transformer(text) \ No newline at end of file + return transformer(text)