v0.4.3 - Upscale OOM fixes, Background Extractor optimization bugfix

saddam213 · saddam213 · commit 53a124269480 · 2026-01-25T13:13:08.000+13:00
diff --git a/DiffuseApp/DiffuseApp.Common/CSnakes.Runtime.dll b/DiffuseApp/DiffuseApp.Common/CSnakes.Runtime.dll
diff --git a/DiffuseApp/DiffuseApp.Server/CSnakes.Runtime.dll b/DiffuseApp/DiffuseApp.Server/CSnakes.Runtime.dll
diff --git a/DiffuseApp/DiffuseApp/CSnakes.Runtime.dll b/DiffuseApp/DiffuseApp/CSnakes.Runtime.dll
diff --git a/DiffuseApp/DiffuseApp/Controls/ModelControl.xaml.cs b/DiffuseApp/DiffuseApp/Controls/ModelControl.xaml.cs
@@ -153,18 +153,9 @@ private bool CanLoad()
 
         private Task UnloadAsync()
         {
-            SelectedExtractor = default;
             _currentExtractor = default;
-
-            SelectedUpscaler = default;
             _currentUpscaler = default;
-
-            CurrentPipeline = new PipelineModel
-            {
-                Device = _selectedDevice
-            };
-
-            SelectionChanged?.Invoke(this, CurrentPipeline);
+            SelectionChanged?.Invoke(this, default);
             return Task.CompletedTask;
         }
 
diff --git a/DiffuseApp/DiffuseApp/Dialogs/MediaPreviewDialog.xaml.cs b/DiffuseApp/DiffuseApp/Dialogs/MediaPreviewDialog.xaml.cs
@@ -60,23 +60,21 @@ public VideoInputStream CurrentVideoStream
         }
 
 
-        public Task<bool> ShowDialogAsync(IHistoryItem selectedItem)
+        public async Task<bool> ShowDialogAsync(IHistoryItem selectedItem)
         {
             HistoryCollection.MoveCurrentTo(selectedItem);
-            SetCurrentImage();
-            return base.ShowDialogAsync();
+            await SetCurrentImage();
+            return await base.ShowDialogAsync();
         }
 
 
-        private Task PrevAsync()
+        private async Task PrevAsync()
         {
             if (CanMovePrev())
             {
                 HistoryCollection.MoveCurrentToPrevious();
-                SetCurrentImage();
+                await SetCurrentImage();
             }
-
-            return Task.CompletedTask;
         }
 
 
diff --git a/DiffuseApp/DiffuseApp/Services/ExtractService.cs b/DiffuseApp/DiffuseApp/Services/ExtractService.cs
@@ -117,7 +117,7 @@ public async Task LoadAsync(PipelineModel pipeline)
                         SampleSize = model.SampleSize
                     };
 
-                    _currentConfig.SetProvider(device.GetProvider());
+                    _currentConfig.SetProvider(device.GetProvider(Microsoft.ML.OnnxRuntime.GraphOptimizationLevel.ORT_DISABLE_ALL));
                     _extractPipeline = model.Type switch
                     {
                         ExtractorType.Pose => PosePipeline.Create(_currentConfig),
diff --git a/DiffuseApp/DiffuseApp/Settings.default.json b/DiffuseApp/DiffuseApp/Settings.default.json
@@ -152,7 +152,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/APISR-GRL-GAN-4x/model.onnx" ]
@@ -204,7 +204,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/RealESRGAN-4x/model.onnx" ]
@@ -217,7 +217,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 1024,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/RealESR-General-4x/model.onnx" ]
@@ -230,7 +230,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 256,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/RealWebPhoto-RGT-4x/model.onnx" ]
@@ -243,7 +243,7 @@
       "ScaleFactor": 2,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/Swin2SR-Classical-2x/model.onnx" ]
@@ -256,7 +256,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/Swin2SR-Classical-4x/model.onnx" ]
@@ -269,7 +269,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 256,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/Swin2SR-RealWorld-BSRGAN-PSN-4x/model.onnx" ]
@@ -282,7 +282,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/SwinIR-BSRGAN-4x/model.onnx" ]
@@ -294,7 +294,7 @@
       "SampleSize": 0,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "ScaleFactor": 4,
@@ -308,7 +308,7 @@
       "ScaleFactor": 4,
       "DefaultOptions": {
         "TileMode": "ClipBlend",
-        "TileSize": 512,
+        "TileSize": 384,
         "TileOverlap": 16
       },
       "UrlPaths": [ "https://huggingface.co/TensorStack/Diffuse/resolve/main/Upscale/UltraSharp-4x/model.onnx" ]
diff --git a/DiffuseApp/DiffuseApp/Views/ImageExtractView.xaml b/DiffuseApp/DiffuseApp/Views/ImageExtractView.xaml
@@ -96,7 +96,7 @@
                         <Grid Margin="10,10,10,2" >
                             <CommonControls:ImageElement 
                                 SplitterVisibility="Auto"
-                                SplitterPosition="Center"
+                                SplitterPosition="Source"
                                 SplitterDirection="LeftToRight"
                                 Source="{Binding ResultImage, Mode=TwoWay}" 
                                 OverlaySource="{Binding CompareImage, Mode=TwoWay}" 
diff --git a/DiffuseApp/DiffuseApp/Views/ImageExtractView.xaml.cs b/DiffuseApp/DiffuseApp/Views/ImageExtractView.xaml.cs
@@ -5,7 +5,6 @@
 using System.Diagnostics;
 using System.IO;
 using System.Threading.Tasks;
-
 using TensorStack.Image;
 using TensorStack.WPF;
 using TensorStack.WPF.Controls;
@@ -31,22 +30,6 @@ public ImageExtractView(Settings settings, NavigationService navigationService,
             ExtractService = extractService;
             ExecuteCommand = new AsyncRelayCommand(ExecuteAsync, CanExecute);
             CancelCommand = new AsyncRelayCommand(CancelAsync, CanCancel);
-            Options = new ExtractInputOptions
-            {
-                TileSize = 512,
-                TileOverlap = 16,
-                IsInverted = false,
-                MergeInput = false,
-                Mode = TensorStack.Extractors.Common.BackgroundMode.RemoveBackground,
-                Detections = 0,
-                BodyConfidence = 0.4f,
-                JointConfidence = 0.1f,
-                ColorAlpha = 0.8f,
-                JointRadius = 7f,
-                BoneRadius = 8f,
-                BoneThickness = 1f,
-                IsTransparent = false
-            };
             InitializeComponent();
         }
 
@@ -223,14 +206,21 @@ private bool CanCancel()
 
         protected async void SelectedExtractorChanged(object sender, PipelineModel pipeline)
         {
-            if (pipeline.ExtractModel is not null && !pipeline.ExtractModel.IsValid)
+            if (pipeline?.ExtractModel == null)
             {
-                if (!await pipeline.ExtractModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Extract")))
-                    CurrentPipeline = default;
+                await UnloadPipelineAsync();
             }
+            else
+            {
+                if (pipeline.ExtractModel is not null && !pipeline.ExtractModel.IsValid)
+                {
+                    if (!await pipeline.ExtractModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Extract")))
+                        CurrentPipeline = default;
+                }
 
-            if (CurrentPipeline is not null)
-                await LoadPipelineAsync();
+                if (CurrentPipeline is not null)
+                    await LoadPipelineAsync();
+            }
         }
 
     }
diff --git a/DiffuseApp/DiffuseApp/Views/ImageUpscaleView.xaml.cs b/DiffuseApp/DiffuseApp/Views/ImageUpscaleView.xaml.cs
@@ -208,14 +208,21 @@ private bool CanCancel()
 
         protected async void SelectedUpscalerChanged(object sender, PipelineModel pipeline)
         {
-            if (pipeline.UpscaleModel is not null && !pipeline.UpscaleModel.IsValid)
+            if (pipeline?.UpscaleModel == null)
             {
-                if (!await pipeline.UpscaleModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Upscale")))
-                    CurrentPipeline = default;
+                await UnloadPipelineAsync();
             }
+            else
+            {
+                if (pipeline.UpscaleModel is not null && !pipeline.UpscaleModel.IsValid)
+                {
+                    if (!await pipeline.UpscaleModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Upscale")))
+                        CurrentPipeline = default;
+                }
 
-            if (CurrentPipeline is not null)
-                await LoadPipelineAsync();
+                if (CurrentPipeline is not null)
+                    await LoadPipelineAsync();
+            }
         }
     }
 }
diff --git a/DiffuseApp/DiffuseApp/Views/VideoExtractView.xaml.cs b/DiffuseApp/DiffuseApp/Views/VideoExtractView.xaml.cs
@@ -204,14 +204,21 @@ private bool CanCancel()
 
         protected async void SelectedExtractorChanged(object sender, PipelineModel pipeline)
         {
-            if (pipeline.ExtractModel is not null && !pipeline.ExtractModel.IsValid)
+            if (pipeline?.ExtractModel == null)
             {
-                if (!await pipeline.ExtractModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Extract")))
-                    CurrentPipeline = default;
+                await UnloadPipelineAsync();
             }
+            else
+            {
+                if (pipeline.ExtractModel is not null && !pipeline.ExtractModel.IsValid)
+                {
+                    if (!await pipeline.ExtractModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Extract")))
+                        CurrentPipeline = default;
+                }
 
-            if (CurrentPipeline is not null)
-                await LoadPipelineAsync();
+                if (CurrentPipeline is not null)
+                    await LoadPipelineAsync();
+            }
         }
     }
 }
diff --git a/DiffuseApp/DiffuseApp/Views/VideoInterpolateView.xaml.cs b/DiffuseApp/DiffuseApp/Views/VideoInterpolateView.xaml.cs
@@ -190,7 +190,14 @@ private bool CanCancel()
 
         protected async void SelectedInterpolationChanged(object sender, PipelineModel pipeline)
         {
-            await LoadPipelineAsync();
+            if (pipeline == null)
+            {
+                await UnloadPipelineAsync();
+            }
+            else
+            {
+                await LoadPipelineAsync();
+            }
         }
     }
 }
diff --git a/DiffuseApp/DiffuseApp/Views/VideoUpscaleView.xaml.cs b/DiffuseApp/DiffuseApp/Views/VideoUpscaleView.xaml.cs
@@ -209,14 +209,21 @@ private bool CanCancel()
 
         protected async void SelectedUpscalerChanged(object sender, PipelineModel pipeline)
         {
-            if (pipeline.UpscaleModel is not null && !pipeline.UpscaleModel.IsValid)
+            if (pipeline?.UpscaleModel == null)
             {
-                if (!await pipeline.UpscaleModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Upscale")))
-                    CurrentPipeline = default;
+                await UnloadPipelineAsync();
             }
+            else
+            {
+                if (pipeline.UpscaleModel is not null && !pipeline.UpscaleModel.IsValid)
+                {
+                    if (!await pipeline.UpscaleModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Upscale")))
+                        CurrentPipeline = default;
+                }
 
-            if (CurrentPipeline is not null)
-                await LoadPipelineAsync();
+                if (CurrentPipeline is not null)
+                    await LoadPipelineAsync();
+            }
         }
     }
 }
diff --git a/Directory.Build.props b/Directory.Build.props
@@ -1,6 +1,6 @@
 <Project>
   <PropertyGroup>
-    <Version>0.4.0</Version>
+    <Version>0.4.3</Version>
     <Company>TensorStack</Company>
     <Copyright>TensorStack - 2026</Copyright>
     <RepositoryUrl>https://github.com/TensorStack-AI/Diffuse</RepositoryUrl>

Original file line number	Diff line number	Diff line change
`@@ -60,23 +60,21 @@ public VideoInputStream CurrentVideoStream`
`60`	`60`	`}`
`61`	`61`
`62`	`62`
`63`		`- public Task<bool> ShowDialogAsync(IHistoryItem selectedItem)`
	`63`	`+ public async Task<bool> ShowDialogAsync(IHistoryItem selectedItem)`
`64`	`64`	`{`
`65`	`65`	`HistoryCollection.MoveCurrentTo(selectedItem);`
`66`		`- SetCurrentImage();`
`67`		`- return base.ShowDialogAsync();`
	`66`	`+ await SetCurrentImage();`
	`67`	`+ return await base.ShowDialogAsync();`
`68`	`68`	`}`
`69`	`69`
`70`	`70`
`71`		`- private Task PrevAsync()`
	`71`	`+ private async Task PrevAsync()`
`72`	`72`	`{`
`73`	`73`	`if (CanMovePrev())`
`74`	`74`	`{`
`75`	`75`	`HistoryCollection.MoveCurrentToPrevious();`
`76`		`- SetCurrentImage();`
	`76`	`+ await SetCurrentImage();`
`77`	`77`	`}`
`78`		`-`
`79`		`- return Task.CompletedTask;`
`80`	`78`	`}`
`81`	`79`
`82`	`80`
Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,7 @@ public async Task LoadAsync(PipelineModel pipeline)`
`117`	`117`	`SampleSize = model.SampleSize`
`118`	`118`	`};`
`119`	`119`
`120`		`- _currentConfig.SetProvider(device.GetProvider());`
	`120`	`+ _currentConfig.SetProvider(device.GetProvider(Microsoft.ML.OnnxRuntime.GraphOptimizationLevel.ORT_DISABLE_ALL));`
`121`	`121`	`_extractPipeline = model.Type switch`
`122`	`122`	`{`
`123`	`123`	`ExtractorType.Pose => PosePipeline.Create(_currentConfig),`
Original file line number	Diff line number	Diff line change
`@@ -208,14 +208,21 @@ private bool CanCancel()`
`208`	`208`
`209`	`209`	`protected async void SelectedUpscalerChanged(object sender, PipelineModel pipeline)`
`210`	`210`	`{`
`211`		`- if (pipeline.UpscaleModel is not null && !pipeline.UpscaleModel.IsValid)`
	`211`	`+ if (pipeline?.UpscaleModel == null)`
`212`	`212`	`{`
`213`		`- if (!await pipeline.UpscaleModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Upscale")))`
`214`		`- CurrentPipeline = default;`
	`213`	`+ await UnloadPipelineAsync();`
`215`	`214`	`}`
	`215`	`+ else`
	`216`	`+ {`
	`217`	`+ if (pipeline.UpscaleModel is not null && !pipeline.UpscaleModel.IsValid)`
	`218`	`+ {`
	`219`	`+ if (!await pipeline.UpscaleModel.DownloadAsync(Path.Combine(Settings.DirectoryModel, "Upscale")))`
	`220`	`+ CurrentPipeline = default;`
	`221`	`+ }`
`216`	`222`
`217`		`- if (CurrentPipeline is not null)`
`218`		`- await LoadPipelineAsync();`
	`223`	`+ if (CurrentPipeline is not null)`
	`224`	`+ await LoadPipelineAsync();`
	`225`	`+ }`
`219`	`226`	`}`
`220`	`227`	`}`
`221`	`228`	`}`