StableDiffusion.NET/StableDiffusion.NET/Models/Builder/ESRGANModelBuilder.cs

using JetBrains.Annotations;

namespace StableDiffusion.NET;

[PublicAPI]
public sealed class ESRGANModelBuilder : IUpscaleModelBuilder
{
    #region Properties & Fields

    public UpscaleModelParameter Parameter { get; }
    IUpscaleModelParameter IUpscaleModelBuilder.Parameter => Parameter;
    IModelParameter IModelBuilder.Parameter => Parameter;

    #endregion

    #region Constructors

    public ESRGANModelBuilder(string modelPath)
    {
        Parameter = new UpscaleModelParameter { ModelPath = modelPath };
    }

    #endregion

    #region Methods

    public UpscaleModel Build() => new(Parameter);

    #endregion
}