Compare commits

..

No commits in common. "631c8ca18074c1b69dfdd001a6349ba8e6b41748" and "037bdab8aa8e99386b90ae1f116bfc30d75cf91b" have entirely different histories.

6 changed files with 7 additions and 39 deletions

3
.gitignore vendored
View File

@ -253,6 +253,3 @@ paket-files/
# JetBrains Rider # JetBrains Rider
.idea/ .idea/
*.sln.iml *.sln.iml
# Test-Project
SD_Test/

View File

@ -13,10 +13,7 @@ public class ModelParameter
public string VaePath { get; set; } = string.Empty; public string VaePath { get; set; } = string.Empty;
public string ControlNetPath { get; set; } = string.Empty; public string ControlNetPath { get; set; } = string.Empty;
public string EmbeddingsDirectory { get; set; } = string.Empty; public string EmbeddingsDirectory { get; set; } = string.Empty;
public string StackedIdEmbeddingsDirectory { get; set; } = string.Empty;
public bool KeepControlNetOnCPU { get; set; } = false; public bool KeepControlNetOnCPU { get; set; } = false;
public bool KeepClipOnCPU { get; set; } = false;
public bool KeepVaeOnCPU { get; set; } = false;
//TODO DarthAffe 01.01.2024: K-Quants doesn't seem to work so far //TODO DarthAffe 01.01.2024: K-Quants doesn't seem to work so far
public Quantization Quantization { get; set; } = Quantization.F16; public Quantization Quantization { get; set; } = Quantization.F16;

View File

@ -55,7 +55,6 @@ internal unsafe partial class Native
[MarshalAs(UnmanagedType.LPStr)] string control_net_path_c_str, [MarshalAs(UnmanagedType.LPStr)] string control_net_path_c_str,
[MarshalAs(UnmanagedType.LPStr)] string lora_model_dir, [MarshalAs(UnmanagedType.LPStr)] string lora_model_dir,
[MarshalAs(UnmanagedType.LPStr)] string embed_dir_c_str, [MarshalAs(UnmanagedType.LPStr)] string embed_dir_c_str,
[MarshalAs(UnmanagedType.LPStr)] string stacked_id_embed_dir_c_str,
[MarshalAs(UnmanagedType.I1)] bool vae_decode_only, [MarshalAs(UnmanagedType.I1)] bool vae_decode_only,
[MarshalAs(UnmanagedType.I1)] bool vae_tiling, [MarshalAs(UnmanagedType.I1)] bool vae_tiling,
[MarshalAs(UnmanagedType.I1)] bool free_params_immediately, [MarshalAs(UnmanagedType.I1)] bool free_params_immediately,
@ -63,9 +62,7 @@ internal unsafe partial class Native
sd_type_t wtype, sd_type_t wtype,
rng_type_t rng_type, rng_type_t rng_type,
schedule_t s, schedule_t s,
[MarshalAs(UnmanagedType.I1)] bool keep_clip_on_cpu, [MarshalAs(UnmanagedType.I1)] bool keep_control_net_cpu);
[MarshalAs(UnmanagedType.I1)] bool keep_control_net_cpu,
[MarshalAs(UnmanagedType.I1)] bool keep_vae_on_cpu);
[LibraryImport(LIB_NAME, EntryPoint = "free_sd_ctx")] [LibraryImport(LIB_NAME, EntryPoint = "free_sd_ctx")]
internal static partial void free_sd_ctx(sd_ctx_t* sd_ctx); internal static partial void free_sd_ctx(sd_ctx_t* sd_ctx);
@ -83,10 +80,7 @@ internal unsafe partial class Native
long seed, long seed,
int batch_count, int batch_count,
sd_image_t* control_cond, sd_image_t* control_cond,
float control_strength, float control_strength);
float style_strength,
[MarshalAs(UnmanagedType.I1)] bool normalize_input,
[MarshalAs(UnmanagedType.LPStr)] string input_id_images_path);
[LibraryImport(LIB_NAME, EntryPoint = "img2img")] [LibraryImport(LIB_NAME, EntryPoint = "img2img")]
internal static partial sd_image_t* img2img(sd_ctx_t* sd_ctx, internal static partial sd_image_t* img2img(sd_ctx_t* sd_ctx,

View File

@ -56,7 +56,6 @@ public sealed unsafe class StableDiffusionModel : IDisposable
_parameter.ControlNetPath, _parameter.ControlNetPath,
_parameter.LoraModelDir, _parameter.LoraModelDir,
_parameter.EmbeddingsDirectory, _parameter.EmbeddingsDirectory,
_parameter.StackedIdEmbeddingsDirectory,
_parameter.VaeDecodeOnly, _parameter.VaeDecodeOnly,
_parameter.VaeTiling, _parameter.VaeTiling,
false, false,
@ -64,9 +63,7 @@ public sealed unsafe class StableDiffusionModel : IDisposable
_parameter.Quantization, _parameter.Quantization,
_parameter.RngType, _parameter.RngType,
_parameter.Schedule, _parameter.Schedule,
_parameter.KeepClipOnCPU, _parameter.KeepControlNetOnCPU);
_parameter.KeepControlNetOnCPU,
_parameter.KeepVaeOnCPU);
if (_ctx == null) throw new NullReferenceException("Failed to initialize Stable Diffusion"); if (_ctx == null) throw new NullReferenceException("Failed to initialize Stable Diffusion");
if (_upscalerParameter != null) if (_upscalerParameter != null)
@ -117,10 +114,7 @@ public sealed unsafe class StableDiffusionModel : IDisposable
parameter.Seed, parameter.Seed,
1, 1,
&controlNetImage, &controlNetImage,
parameter.ControlNet.Strength, parameter.ControlNet.Strength);
parameter.PhotoMaker.StyleRatio,
parameter.PhotoMaker.NormalizeInput,
parameter.PhotoMaker.InputIdImageDirectory);
Marshal.FreeHGlobal((nint)controlNetImage.data); Marshal.FreeHGlobal((nint)controlNetImage.data);
} }
@ -146,10 +140,7 @@ public sealed unsafe class StableDiffusionModel : IDisposable
parameter.Seed, parameter.Seed,
1, 1,
&controlNetImage, &controlNetImage,
parameter.ControlNet.Strength, parameter.ControlNet.Strength);
parameter.PhotoMaker.StyleRatio,
parameter.PhotoMaker.NormalizeInput,
parameter.PhotoMaker.InputIdImageDirectory);
} }
} }
} }
@ -167,10 +158,7 @@ public sealed unsafe class StableDiffusionModel : IDisposable
parameter.Seed, parameter.Seed,
1, 1,
null, null,
0, 0);
parameter.PhotoMaker.StyleRatio,
parameter.PhotoMaker.NormalizeInput,
parameter.PhotoMaker.InputIdImageDirectory);
} }
return new StableDiffusionImage(result); return new StableDiffusionImage(result);

View File

@ -15,7 +15,6 @@ public sealed class StableDiffusionParameter
public int ClipSkip { get; set; } = -1; public int ClipSkip { get; set; } = -1;
public StableDiffusionControlNetParameter ControlNet { get; } = new(); public StableDiffusionControlNetParameter ControlNet { get; } = new();
public PhotoMakerParameter PhotoMaker { get; } = new();
#endregion #endregion
} }
@ -32,11 +31,4 @@ public sealed class StableDiffusionControlNetParameter
public float CannyWeak { get; set; } = 0.8f; public float CannyWeak { get; set; } = 0.8f;
public float CannyStrong { get; set; } = 1.0f; public float CannyStrong { get; set; } = 1.0f;
public bool CannyInverse { get; set; } = false; public bool CannyInverse { get; set; } = false;
}
public sealed class PhotoMakerParameter
{
public string InputIdImageDirectory { get; set; } = string.Empty;
public float StyleRatio { get; set; } = 20f;
public bool NormalizeInput { get; set; } = false;
} }

View File

@ -4,7 +4,7 @@ if not exist stable-diffusion.cpp (
cd stable-diffusion.cpp cd stable-diffusion.cpp
git fetch git fetch
git checkout a469688e30122d3b6c1faa5b36ffc3261e6deb82 git checkout 1ce9470f27d480c6aa5d43c0af5b60db99454252
git submodule init git submodule init
git submodule update git submodule update