Rename ProcessType to DiffuserType

Author: saddam213

OnnxStack.StableDiffusion/Common/IModelOptions.cs

@@ -15,7 +15,7 @@ public interface IModelOptions : IOnnxModel
         int TokenizerLimit { get; set; }
         int InputTokenLimit { get; set; }
         int EmbeddingsLength { get; set; }
-        List<ProcessType> Diffusers { get; set; }
+        List<DiffuserType> Diffusers { get; set; }
         ImmutableArray<int> BlankTokenValueArray { get; set; }
     }
 }

OnnxStack.StableDiffusion/Config/ModelOptions.cs

@@ -19,7 +19,7 @@ public class ModelOptions : IModelOptions, IOnnxModelSetConfig
         public int TokenizerLimit { get; set; }
         public int EmbeddingsLength { get; set; }
         public float ScaleFactor { get; set; }
-        public List<ProcessType> Diffusers { get; set; } = new List<ProcessType>();
+        public List<DiffuserType> Diffusers { get; set; } = new List<DiffuserType>();
 
         public int DeviceId { get; set; }
         public int InterOpNumThreads { get; set; }

OnnxStack.StableDiffusion/Config/PromptOptions.cs

@@ -6,7 +6,7 @@ namespace OnnxStack.StableDiffusion.Config
 {
     public class PromptOptions
     {
-        public ProcessType ProcessType { get; set; }
+        public DiffuserType DiffuserType { get; set; }
 
         [Required]
         [StringLength(512, MinimumLength = 4)]
@@ -24,7 +24,7 @@ public class PromptOptions
         public bool HasInputImageMask => InputImageMask?.HasImage ?? false;
     }
 
-    public enum ProcessType
+    public enum DiffuserType
     {
         TextToImage = 0,
         ImageToImage = 1,

OnnxStack.StableDiffusion/Config/StableDiffusionConfig.cs

@@ -24,8 +24,14 @@ public void Initialize()
             foreach (var modelSet in OnnxModelSets)
             {
                 modelSet.InitBlankTokenArray();
-                foreach (var model in modelSet.ModelConfigurations.Where(x => x.Type == OnnxModelType.Tokenizer && string.IsNullOrEmpty(x.OnnxModelPath)))
-                    model.OnnxModelPath = defaultTokenizer;
+                foreach (var model in modelSet.ModelConfigurations)
+                {
+                    if (model.Type == OnnxModelType.Tokenizer && string.IsNullOrEmpty(model.OnnxModelPath))
+                        model.OnnxModelPath = defaultTokenizer;
+
+                    if (!File.Exists(model.OnnxModelPath))
+                        modelSet.IsEnabled = false;
+                }
             }
         }
     }

OnnxStack.StableDiffusion/Services/StableDiffusionService.cs

@@ -146,12 +146,12 @@ public async Task<Stream> GenerateAsStreamAsync(IModelOptions model, PromptOptio
 
         private async Task<DenseTensor<float>> RunAsync(IModelOptions modelOptions, PromptOptions promptOptions, SchedulerOptions schedulerOptions, Action<int, int> progress = null, CancellationToken cancellationToken = default)
         {
-            return promptOptions.ProcessType switch
+            return promptOptions.DiffuserType switch
             {
-                ProcessType.TextToImage => await _textDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
-                ProcessType.ImageToImage => await _imageDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
-                ProcessType.ImageInpaint => await _inpaintDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
-                ProcessType.ImageInpaintLegacy => await _inpaintLegacyDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
+                DiffuserType.TextToImage => await _textDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
+                DiffuserType.ImageToImage => await _imageDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
+                DiffuserType.ImageInpaint => await _inpaintDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
+                DiffuserType.ImageInpaintLegacy => await _inpaintLegacyDiffuser.DiffuseAsync(modelOptions, promptOptions, schedulerOptions, progress, cancellationToken),
                 _ => throw new NotImplementedException()
             };
         }

OnnxStack.WebUI/Hubs/StableDiffusionHub.cs

@@ -126,7 +126,7 @@ public async IAsyncEnumerable<StableDiffusionResult> OnExecuteImageInpaint(Promp
         private async Task<StableDiffusionResult> GenerateTextToImageResult(PromptOptions promptOptions, SchedulerOptions schedulerOptions, CancellationToken cancellationToken)
         {
             var timestamp = Stopwatch.GetTimestamp();
-            promptOptions.ProcessType = ProcessType.TextToImage;
+            promptOptions.DiffuserType = DiffuserType.TextToImage;
             schedulerOptions.Seed = GenerateSeed(schedulerOptions.Seed);
 
             //1. Create filenames
@@ -166,7 +166,7 @@ private async Task<StableDiffusionResult> GenerateTextToImageResult(PromptOption
         private async Task<StableDiffusionResult> GenerateImageToImageResult(PromptOptions promptOptions, SchedulerOptions schedulerOptions, CancellationToken cancellationToken)
         {
             var timestamp = Stopwatch.GetTimestamp();
-            promptOptions.ProcessType = ProcessType.ImageToImage;
+            promptOptions.DiffuserType = DiffuserType.ImageToImage;
             schedulerOptions.Seed = GenerateSeed(schedulerOptions.Seed);
 
             //1. Create filenames
@@ -213,7 +213,7 @@ private async Task<StableDiffusionResult> GenerateImageToImageResult(PromptOptio
         private async Task<StableDiffusionResult> GenerateImageInpaintResult(PromptOptions promptOptions, SchedulerOptions schedulerOptions, CancellationToken cancellationToken)
         {
             var timestamp = Stopwatch.GetTimestamp();
-            promptOptions.ProcessType = ProcessType.ImageInpaint;
+            promptOptions.DiffuserType = DiffuserType.ImageInpaint;
             schedulerOptions.Seed = GenerateSeed(schedulerOptions.Seed);
 
             // 1. Create filenames

OnnxStack.WebUI/Models/ImageBlueprint.cs

@@ -15,14 +15,14 @@ public ImageBlueprint(PromptOptions prompt, SchedulerOptions options, string out
             Prompt = prompt.Prompt;
             NegativePrompt = prompt.NegativePrompt;
             SchedulerType = prompt.SchedulerType;
-            ProcessType = prompt.ProcessType;
+            DiffuserType = prompt.DiffuserType;
         }
 
         public DateTime Timestamp { get; }
         public string Prompt { get; }
         public string NegativePrompt { get; }
         public SchedulerType SchedulerType { get; }
-        public ProcessType ProcessType { get; }
+        public DiffuserType DiffuserType { get; }
         public string MaskImageUrl { get; }
         public string InputImageUrl { get; }
         public string OutputImageUrl { get; }