@@ -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