v5.0

MitchellShibilski-Unkel · Jul 25, 2024 · 821c940 · 821c940
1 parent 1d0d695
commit 821c940
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)
+            return transformer(text)