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

Commit 74834fc

Browse files
committed
Fix UI build
1 parent 3024601 commit 74834fc

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

OnnxStack.UI/Models/UpdateControlNetModelSetViewModel.cs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,8 @@ public static UpdateControlNetModelSetViewModel FromModelSet(ControlNetModelSet
8585
return new UpdateControlNetModelSetViewModel
8686
{
8787
Name = modelset.Name,
88-
ControlNetType = modelset.Type,
89-
PipelineType = modelset.PipelineType,
88+
ControlNetType = modelset.ControlNetConfig.Type,
89+
PipelineType = modelset.ControlNetConfig.PipelineType,
9090
DeviceId = modelset.DeviceId,
9191
ExecutionMode = modelset.ExecutionMode,
9292
ExecutionProvider = modelset.ExecutionProvider,
@@ -102,15 +102,16 @@ public static ControlNetModelSet ToModelSet(UpdateControlNetModelSetViewModel mo
102102
{
103103
IsEnabled = true,
104104
Name = modelset.Name,
105-
Type = modelset.ControlNetType,
106-
PipelineType = modelset.PipelineType,
105+
107106
DeviceId = modelset.DeviceId,
108107
ExecutionMode = modelset.ExecutionMode,
109108
ExecutionProvider = modelset.ExecutionProvider,
110109
InterOpNumThreads = modelset.InterOpNumThreads,
111110
IntraOpNumThreads = modelset.IntraOpNumThreads,
112111
ControlNetConfig = new ControlNetModelConfig
113112
{
113+
Type = modelset.ControlNetType,
114+
PipelineType = modelset.PipelineType,
114115
OnnxModelPath = modelset.ModelFile
115116
}
116117
};

OnnxStack.UI/Services/ModelFactory.cs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,8 +149,7 @@ public ControlNetModelSet CreateControlNetModelSet(string name, ControlNetType c
149149
return new ControlNetModelSet
150150
{
151151
Name = name,
152-
Type = controlNetType,
153-
PipelineType = pipelineType,
152+
154153
IsEnabled = true,
155154
DeviceId = _settings.DefaultDeviceId,
156155
ExecutionMode = _settings.DefaultExecutionMode,
@@ -159,6 +158,8 @@ public ControlNetModelSet CreateControlNetModelSet(string name, ControlNetType c
159158
IntraOpNumThreads = _settings.DefaultIntraOpNumThreads,
160159
ControlNetConfig = new ControlNetModelConfig
161160
{
161+
Type = controlNetType,
162+
PipelineType = pipelineType,
162163
OnnxModelPath = modelFilename
163164
}
164165
};

0 commit comments

Comments
 (0)