From 99ee47b79b2a651a640576985e7d5da7e1b7b5d2 Mon Sep 17 00:00:00 2001 From: dunkeroni Date: Wed, 6 Sep 2023 09:39:47 -0400 Subject: [PATCH] Added square root function --- invokeai/app/invocations/math.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/invokeai/app/invocations/math.py b/invokeai/app/invocations/math.py index e3da88e4bf..fa10a20f8b 100644 --- a/invokeai/app/invocations/math.py +++ b/invokeai/app/invocations/math.py @@ -146,4 +146,14 @@ class ModulusInvocation(BaseInvocation): b: int = InputField(default=0, description=FieldDescriptions.num_2) def invoke(self, context: InvocationContext) -> IntegerOutput: - return IntegerOutput(value=self.a % self.b) \ No newline at end of file + return IntegerOutput(value=self.a % self.b) + + +@invocation("sqrt", title="Square Root", tags=["math", "sqrt"], category="math", version="1.0.0") +class SquareRootInvocation(BaseInvocation): + """Returns the square root of a number.""" + + value: float = InputField(default=0, ge=0, description="The float value") + + def invoke(self, context: InvocationContext) -> FloatOutput: + return FloatOutput(value=np.sqrt(self.value)) \ No newline at end of file