You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.1 KiB
36 lines
1.1 KiB
using StabilityMatrix.Avalonia.Languages; |
|
using StabilityMatrix.Avalonia.Models.Inference; |
|
using StabilityMatrix.Avalonia.Services; |
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
|
using StabilityMatrix.Core.Attributes; |
|
using StabilityMatrix.Core.Models.Api.Comfy.Nodes; |
|
|
|
namespace StabilityMatrix.Avalonia.ViewModels.Inference.Modules; |
|
|
|
[ManagedService] |
|
[Transient] |
|
public class SaveImageModule : ModuleBase |
|
{ |
|
/// <inheritdoc /> |
|
public SaveImageModule(ServiceManager<ViewModelBase> vmFactory) |
|
: base(vmFactory) |
|
{ |
|
Title = Resources.Label_SaveIntermediateImage; |
|
} |
|
|
|
/// <inheritdoc /> |
|
protected override void OnApplyStep(ModuleApplyStepEventArgs e) |
|
{ |
|
var preview = e.Builder |
|
.Nodes |
|
.AddTypedNode( |
|
new ComfyNodeBuilder.PreviewImage |
|
{ |
|
Name = e.Builder.Nodes.GetUniqueName("SaveIntermediateImage"), |
|
Images = e.Builder.GetPrimaryAsImage() |
|
} |
|
); |
|
|
|
e.Builder.Connections.OutputNodes.Add(preview); |
|
} |
|
}
|
|
|