diff --git a/ui/index.html b/ui/index.html
index 8cfde5f9..ed35db3c 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -15,7 +15,7 @@
-
Stable Diffusion UI v2.19
+ Stable Diffusion UI v2.195
-
diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py
index eb456c36..d492b269 100644
--- a/ui/sd_internal/runtime.py
+++ b/ui/sd_internal/runtime.py
@@ -444,7 +444,7 @@ def do_mk_img(req: Request):
if return_orig_img:
save_image(img, img_out_path)
- save_metadata(meta_out_path, prompts, opt_seed, opt_W, opt_H, opt_ddim_steps, opt_scale, opt_strength, opt_use_face_correction, opt_use_upscale, opt_sampler_name)
+ save_metadata(meta_out_path, prompts, opt_seed, opt_W, opt_H, opt_ddim_steps, opt_scale, opt_strength, opt_use_face_correction, opt_use_upscale, opt_sampler_name, req.negative_prompt)
if return_orig_img:
img_data = img_to_base64_str(img)
@@ -505,8 +505,8 @@ def save_image(img, img_out_path):
except:
print('could not save the file', traceback.format_exc())
-def save_metadata(meta_out_path, prompts, opt_seed, opt_W, opt_H, opt_ddim_steps, opt_scale, opt_prompt_strength, opt_correct_face, opt_upscale, sampler_name):
- metadata = f"{prompts[0]}\nWidth: {opt_W}\nHeight: {opt_H}\nSeed: {opt_seed}\nSteps: {opt_ddim_steps}\nGuidance Scale: {opt_scale}\nPrompt Strength: {opt_prompt_strength}\nUse Face Correction: {opt_correct_face}\nUse Upscaling: {opt_upscale}\nSampler: {sampler_name}"
+def save_metadata(meta_out_path, prompts, opt_seed, opt_W, opt_H, opt_ddim_steps, opt_scale, opt_prompt_strength, opt_correct_face, opt_upscale, sampler_name, negative_prompt):
+ metadata = f"{prompts[0]}\nWidth: {opt_W}\nHeight: {opt_H}\nSeed: {opt_seed}\nSteps: {opt_ddim_steps}\nGuidance Scale: {opt_scale}\nPrompt Strength: {opt_prompt_strength}\nUse Face Correction: {opt_correct_face}\nUse Upscaling: {opt_upscale}\nSampler: {sampler_name}\nNegative Prompt: {negative_prompt}"
try:
with open(meta_out_path, 'w') as f:
diff --git a/ui/server.py b/ui/server.py
index c8c7324f..67c84949 100644
--- a/ui/server.py
+++ b/ui/server.py
@@ -84,7 +84,10 @@ async def ping():
model_is_loading = True
from sd_internal import runtime
- runtime.load_model_ckpt(ckpt_to_use="sd-v1-4")
+
+ custom_weight_path = os.path.join(SCRIPT_DIR, 'custom-model.ckpt')
+ ckpt_to_use = "sd-v1-4" if not os.path.exists(custom_weight_path) else "custom-model"
+ runtime.load_model_ckpt(ckpt_to_use=ckpt_to_use)
model_loaded = True
model_is_loading = False