From 821c94012badc529db96a3f92270b2ffdfd8d838 Mon Sep 17 00:00:00 2001 From: Mitchell Shibilski-Unkel <133730558+MitchellShibilski-Unkel@users.noreply.github.com> Date: Wed, 24 Jul 2024 23:01:19 -0600 Subject: [PATCH] v5.0 --- pyai.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)