Skip to content
This repository was archived by the owner on Nov 27, 2024. It is now read-only.

Commit 08466f6

Browse files
committed
Add Pipleline selector to Settings view
1 parent b15ceb2 commit 08466f6

File tree

3 files changed

+20
-1
lines changed

3 files changed

+20
-1
lines changed

OnnxStack.UI/App.xaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,12 @@
8181
</ObjectDataProvider.MethodParameters>
8282
</ObjectDataProvider>
8383

84+
<ObjectDataProvider x:Key="DiffuserPipelineType" MethodName="GetValues" ObjectType="{x:Type system:Enum}">
85+
<ObjectDataProvider.MethodParameters>
86+
<x:Type TypeName="SD_Enums:DiffuserPipelineType"/>
87+
</ObjectDataProvider.MethodParameters>
88+
</ObjectDataProvider>
89+
8490

8591
<!--TODO: Style dictionary for themeing-->
8692
<Style TargetType="{x:Type TextBox}">

OnnxStack.UI/Views/Settings.xaml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,11 @@
5757
</Grid.ColumnDefinitions>
5858

5959
<StackPanel>
60-
<UniformGrid Columns="3">
60+
<UniformGrid Columns="5">
61+
<StackPanel>
62+
<TextBlock Text="Pipeline" />
63+
<ComboBox ItemsSource="{Binding Source={StaticResource DiffuserPipelineType}}" SelectedItem="{Binding SelectedModelSet.PipelineType}" />
64+
</StackPanel>
6165
<StackPanel VerticalAlignment="Center">
6266
<CheckBox IsChecked="{Binding SelectedModelSet.IsEnabled}" Content="IsEnabled" />
6367
</StackPanel>

OnnxStack.UI/Views/Settings.xaml.cs

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -269,6 +269,7 @@ private ModelSetEditModel CreateEditModel(ModelOptions modelOptions)
269269
PadTokenId = modelOptions.PadTokenId,
270270
ScaleFactor = modelOptions.ScaleFactor,
271271
TokenizerLimit = modelOptions.TokenizerLimit,
272+
PipelineType = modelOptions.PipelineType,
272273
EnableTextToImage = modelOptions.Diffusers.Contains(DiffuserType.TextToImage),
273274
EnableImageToImage = modelOptions.Diffusers.Contains(DiffuserType.ImageToImage),
274275
EnableImageInpaint = modelOptions.Diffusers.Contains(DiffuserType.ImageInpaint) || modelOptions.Diffusers.Contains(DiffuserType.ImageInpaintLegacy),
@@ -310,6 +311,7 @@ private ModelOptions CreateModelOptions(ModelSetEditModel editModel)
310311
PadTokenId = editModel.PadTokenId,
311312
ScaleFactor = editModel.ScaleFactor,
312313
TokenizerLimit = editModel.TokenizerLimit,
314+
PipelineType = editModel.PipelineType,
313315
Diffusers = new List<DiffuserType>(editModel.GetDiffusers()),
314316
ModelConfigurations = new List<OnnxModelSessionConfig>(editModel.ModelConfigurations.Select(x => new OnnxModelSessionConfig
315317
{
@@ -369,6 +371,7 @@ public class ModelSetEditModel : INotifyPropertyChanged
369371
private bool _enableImageToImage;
370372
private bool _enableImageInpaint;
371373
private bool _enableImageInpaintLegacy;
374+
private DiffuserPipelineType _pipelineType;
372375

373376
public string Name
374377
{
@@ -484,6 +487,12 @@ public ObservableCollection<ModelSessionEditModel> ModelConfigurations
484487
set { _modelConfigurations = value; NotifyPropertyChanged(); }
485488
}
486489

490+
public DiffuserPipelineType PipelineType
491+
{
492+
get { return _pipelineType; }
493+
set { _pipelineType = value; NotifyPropertyChanged(); }
494+
}
495+
487496

488497
public IEnumerable<DiffuserType> GetDiffusers()
489498
{

0 commit comments

Comments
 (0)