Skip to content

Commit ef232f9

Browse files
committed
ApplySpecialLogic should be handled more carefully
1 parent 9dc665b commit ef232f9

File tree

4 files changed

+6
-1
lines changed

4 files changed

+6
-1
lines changed

src/BuiltinExtensions/ComfyUIBackend/ComfyUIWebAPI.cs

+1
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,7 @@ public static async Task<JObject> ComfyGetGeneratedWorkflow(Session session, JOb
139139
try
140140
{
141141
input = T2IAPI.RequestToParams(session, rawInput);
142+
input.ApplySpecialLogic();
142143
input.PreparsePromptLikes();
143144
ComfyUIAPIAbstractBackend backend = ComfyUIBackendExtension.ComfyBackendsDirect().FirstOrDefault().Backend as ComfyUIAPIAbstractBackend;
144145
if (backend is null)

src/BuiltinExtensions/GridGenerator/GridGeneratorExtension.cs

+1
Original file line numberDiff line numberDiff line change
@@ -420,6 +420,7 @@ async Task sendStatus()
420420
baseParams.Remove(T2IParamTypes.BatchSize);
421421
baseParams.Remove(T2IParamTypes.Images);
422422
baseParams.Remove(T2IParamTypes.OutputIntermediateImages);
423+
baseParams.ApplySpecialLogic();
423424
await sendStatus();
424425
SwarmUIGridData data = new()
425426
{

src/BuiltinExtensions/ImageBatchTool/ImageBatchToolExtension.cs

+2
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,7 @@ void setRes(int width, int height)
134134
param.Set(T2IParamTypes.Height, height);
135135
param.Remove(T2IParamTypes.AspectRatio);
136136
param.Remove(T2IParamTypes.AltResolutionHeightMult);
137+
param.Remove(T2IParamTypes.RawResolution);
137138
}
138139
switch (resMode)
139140
{
@@ -176,6 +177,7 @@ void setRes(int width, int height)
176177
param.Set(controlnetParams.Image, image);
177178
}
178179
}
180+
param.ApplySpecialLogic();
179181
int genId = 0;
180182
tasks.Add(T2IEngine.CreateImageTask(param, $"{imageIndex}", claim, output, setError, isWS, Program.ServerSettings.Backends.PerRequestTimeoutMinutes, (image, metadata) =>
181183
{

src/WebAPI/T2IAPI.cs

+2-1
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,6 @@ public static T2IParamInput RequestToParams(Session session, JObject rawInput)
221221
}
222222
user_input.ExtraMeta["presets_used"] = presets.Values().Select(v => v.ToString()).ToList();
223223
}
224-
user_input.ApplySpecialLogic();
225224
return user_input;
226225
}
227226

@@ -252,6 +251,7 @@ void setError(string message)
252251
setError(ex.Message);
253252
return;
254253
}
254+
user_input.ApplySpecialLogic();
255255
images = user_input.Get(T2IParamTypes.Images, images);
256256
Logs.Info($"User {session.User.UserID} requested {images} image{(images == 1 ? "" : "s")} with model '{user_input.Get(T2IParamTypes.Model)?.Name}'...");
257257
if (Logs.MinimumLevel <= Logs.LogLevel.Verbose)
@@ -446,6 +446,7 @@ public static async Task<JObject> AddImageToHistory(Session session,
446446
{
447447
return new() { ["error"] = ex.Message };
448448
}
449+
user_input.ApplySpecialLogic();
449450
Logs.Info($"User {session.User.UserID} stored an image to history.");
450451
(img, string metadata) = user_input.SourceSession.ApplyMetadata(img, user_input, 1);
451452
(string path, _) = session.SaveImage(img, 0, user_input, metadata);

0 commit comments

Comments
 (0)