From 4afbcdbca803daa308d6098eeda5dfe986ebcd4d Mon Sep 17 00:00:00 2001
From: Darth Affe <darthaffe@wyrez.org>
Date: Thu, 2 May 2024 23:59:43 +0200
Subject: [PATCH] Improved Parameter validation

---
 StableDiffusion.NET/Extensions/ParameterExtension.cs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/StableDiffusion.NET/Extensions/ParameterExtension.cs b/StableDiffusion.NET/Extensions/ParameterExtension.cs
index 78d6d67..13fd6d8 100644
--- a/StableDiffusion.NET/Extensions/ParameterExtension.cs
+++ b/StableDiffusion.NET/Extensions/ParameterExtension.cs
@@ -9,6 +9,8 @@ public static class ParameterExtension
     public static void Validate(this StableDiffusionParameter parameter)
     {
         ArgumentNullException.ThrowIfNull(parameter, nameof(parameter));
+        ArgumentNullException.ThrowIfNull(parameter.ControlNet, nameof(StableDiffusionParameter.ControlNet));
+        ArgumentNullException.ThrowIfNull(parameter.PhotoMaker, nameof(StableDiffusionParameter.PhotoMaker));
         ArgumentNullException.ThrowIfNull(parameter.NegativePrompt, nameof(StableDiffusionParameter.NegativePrompt));
 
         ArgumentOutOfRangeException.ThrowIfNegativeOrZero(parameter.Width, nameof(StableDiffusionParameter.Width));
@@ -20,10 +22,8 @@ public static class ParameterExtension
 
         if (!Enum.IsDefined(parameter.SampleMethod)) throw new ArgumentOutOfRangeException(nameof(StableDiffusionParameter.SampleMethod));
 
-        // ReSharper disable ConditionalAccessQualifierIsNonNullableAccordingToAPIContract
-        parameter.ControlNet?.Validate();
-        parameter.PhotoMaker?.Validate();
-        // ReSharper restore ConditionalAccessQualifierIsNonNullableAccordingToAPIContract
+        parameter.ControlNet.Validate();
+        parameter.PhotoMaker.Validate();
     }
 
     public static void Validate(this StableDiffusionControlNetParameter parameter)