Fix channel dimension in textprocessing and make dimension overrides EP-independent

Author: keshavv27

AIDevGallery/Samples/SharedCode/StableDiffusionCode/SafetyChecker.cs

@@ -58,13 +58,10 @@ private Task<InferenceSession> GetInferenceSession(string modelPath, ExecutionPr
             SessionOptions sessionOptions = new();
             sessionOptions.RegisterOrtExtensions();
 
-            if(device == "NvTensorRTRTXExecutionProvider")
-            {
-                sessionOptions.AddFreeDimensionOverrideByName("batch", 1);
-                sessionOptions.AddFreeDimensionOverrideByName("channels", 3);
-                sessionOptions.AddFreeDimensionOverrideByName("height", 224);
-                sessionOptions.AddFreeDimensionOverrideByName("width", 224);
-            }
+            sessionOptions.AddFreeDimensionOverrideByName("batch", 1);
+            sessionOptions.AddFreeDimensionOverrideByName("channels", 3);
+            sessionOptions.AddFreeDimensionOverrideByName("height", 224);
+            sessionOptions.AddFreeDimensionOverrideByName("width", 224);
 
             if (policy != null)
             {

AIDevGallery/Samples/SharedCode/StableDiffusionCode/StableDiffusion.cs

@@ -83,15 +83,12 @@ private Task<InferenceSession> GetInferenceSession(string modelPath, ExecutionPr
             SessionOptions sessionOptions = new();
             sessionOptions.RegisterOrtExtensions();
 
-            if(device == "NvTensorRTRTXExecutionProvider")
-            {
-                sessionOptions.AddFreeDimensionOverrideByName("batch", 2);
-                sessionOptions.AddFreeDimensionOverrideByName("time_batch", 1);
-                sessionOptions.AddFreeDimensionOverrideByName("channels", 4);
-                sessionOptions.AddFreeDimensionOverrideByName("height", 64);
-                sessionOptions.AddFreeDimensionOverrideByName("width", 64);
-                sessionOptions.AddFreeDimensionOverrideByName("sequence", 77);
-            }
+            sessionOptions.AddFreeDimensionOverrideByName("batch", 2);
+            sessionOptions.AddFreeDimensionOverrideByName("time_batch", 1);
+            sessionOptions.AddFreeDimensionOverrideByName("channels", 4);
+            sessionOptions.AddFreeDimensionOverrideByName("height", 64);
+            sessionOptions.AddFreeDimensionOverrideByName("width", 64);
+            sessionOptions.AddFreeDimensionOverrideByName("sequence", 77);
 
 
             if (policy != null)

AIDevGallery/Samples/SharedCode/StableDiffusionCode/TextProcessing.cs

@@ -59,14 +59,11 @@ private Task<InferenceSession> GetInferenceSession(string modelPath, ExecutionPr
 
             SessionOptions sessionOptions = new();
             sessionOptions.RegisterOrtExtensions();
-
-            if(device == "NvTensorRTRTXExecutionProvider")
-            {
-                sessionOptions.AddFreeDimensionOverrideByName("batch", 1);
-                sessionOptions.AddFreeDimensionOverrideByName("channels", 4);
-                sessionOptions.AddFreeDimensionOverrideByName("height", 512);
-                sessionOptions.AddFreeDimensionOverrideByName("width", 512);
-            }
+            
+            sessionOptions.AddFreeDimensionOverrideByName("batch", 1);
+            sessionOptions.AddFreeDimensionOverrideByName("channels", 3);
+            sessionOptions.AddFreeDimensionOverrideByName("height", 512);
+            sessionOptions.AddFreeDimensionOverrideByName("width", 512);            
 
             if (policy != null)
             {

AIDevGallery/Samples/SharedCode/StableDiffusionCode/VaeDecoder.cs

@@ -57,13 +57,10 @@ private Task<InferenceSession> GetInferenceSession(string modelPath, ExecutionPr
             SessionOptions sessionOptions = new();
             sessionOptions.RegisterOrtExtensions();
 
-            if(device == "NvTensorRTRTXExecutionProvider")
-            {
-                sessionOptions.AddFreeDimensionOverrideByName("batch", 1);
-                sessionOptions.AddFreeDimensionOverrideByName("channels", 4);
-                sessionOptions.AddFreeDimensionOverrideByName("height", 64);
-                sessionOptions.AddFreeDimensionOverrideByName("width", 64);
-            }            
+            sessionOptions.AddFreeDimensionOverrideByName("batch", 1);
+            sessionOptions.AddFreeDimensionOverrideByName("channels", 4);
+            sessionOptions.AddFreeDimensionOverrideByName("height", 64);
+            sessionOptions.AddFreeDimensionOverrideByName("width", 64);
 
             if (policy != null)
             {