update workflow, use Realistic_Vision and 4x_NMKD-Superscale

app.py

@@ -10,11 +10,14 @@ import subprocess
 
 from huggingface_hub import hf_hub_download
 from huggingface_hub import snapshot_download
-#hf_hub_download(repo_id="black-forest-labs/FLUX.1-Redux-dev", filename="flux1-redux-dev.safetensors", local_dir="models/style_models")
 
-#https://huggingface.co/autismanon/modeldump/blob/main/dreamshaper_8.safetensors
-print("dreamshaper_8.safetensors")
-hf_hub_download(repo_id="autismanon/modeldump", filename="dreamshaper_8.safetensors", local_dir="models/checkpoints/SD1.5/")
+#https://huggingface.co/SG161222/Realistic_Vision_V6.0_B1_noVAE/blob/main/Realistic_Vision_V6.0_NV_B1_fp16.safetensors
+print("Realistic_Vision_V6.0_NV_B1_fp16.safetensors")
+hf_hub_download(repo_id="SG161222/Realistic_Vision_V6.0_B1_noVAE", filename="Realistic_Vision_V6.0_NV_B1_fp16.safetensors", local_dir="models/checkpoints/SD1.5/")
+
+#https://huggingface.co/gemasai/4x_NMKD-Superscale-SP_178000_G/blob/main/4x_NMKD-Superscale-SP_178000_G.pth
+print("4x_NMKD-Superscale-SP_178000_G.pth")
+hf_hub_download(repo_id="gemasai/4x_NMKD-Superscale-SP_178000_G", filename="4x_NMKD-Superscale-SP_178000_G.pth", local_dir="models/upscale_models/")
 
 #https://huggingface.co/lllyasviel/ControlNet-v1-1/blob/main/control_v11p_sd15_openpose.pth
 print("control_v11p_sd15_openpose.pth")
@@ -36,10 +39,6 @@ hf_hub_download(repo_id="naonovn/Lora", filename="add_detail.safetensors", local
 print("BaldifierW2.safetensors")
 hf_hub_download(repo_id="Dreamspire/BaldifierW2", filename="BaldifierW2.safetensors", local_dir="models/loras/")
 
-#https://huggingface.co/lokCX/4x-Ultrasharp/blob/main/4x-UltraSharp.pth
-print("UltraSharp.pth")
-hf_hub_download(repo_id="lokCX/4x-Ultrasharp", filename="4x-UltraSharp.pth", local_dir="models/upscale_models/")
-
 #./clip_vision/CLIP-ViT-H-14-laion2B-s32B-b79K.safetensors
 #https://huggingface.co/h94/IP-Adapter/blob/main/models/image_encoder/model.safetensors
 print("CLIP-ViT-H-14-laion2B-s32B-b79K.safetensors")
@@ -109,7 +108,7 @@ print("ipadapter done")
 
 #./annotator/yzd-v/DWPose/yolox_l.onnx
 
-print("UltraSharp.pth")
+
 
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
     """Returns the value at the given index of a sequence or mapping.
@@ -277,13 +276,13 @@ if "Florence2ModelLoader" in NODE_CLASS_MAPPINGS:
 #5、其它需要提前加载的模型，放到顶层加载
 
 checkpointloadersimple_50 = checkpointloadersimple.load_checkpoint(
-    ckpt_name="SD1.5/dreamshaper_8.safetensors"
+    ckpt_name="SD1.5/Realistic_Vision_V6.0_NV_B1_fp16.safetensors"
 )
 
 
 loraloader_841 = loraloader.load_lora(
     lora_name="add_detail.safetensors",
-    strength_model=2,
+    strength_model=1,
     strength_clip=1,
     model=get_value_at_index(checkpointloadersimple_50, 0),
     clip=get_value_at_index(checkpointloadersimple_50, 1),
@@ -328,7 +327,7 @@ faceanalysismodels_506 = faceanalysismodels.load_models(
 )
 
 upscalemodelloader_835 = upscalemodelloader.load_model(
-    model_name="4x-UltraSharp.pth"
+    model_name="4x_NMKD-Superscale-SP_178000_G.pth"
 )
 
 faceanalysismodels_840 = faceanalysismodels.load_models(
@@ -596,9 +595,9 @@ def generate_image(model_image, hairstyle_template_image):
             seed=random.randint(1, 2**64),
             steps=8,
             cfg=1,
-            sampler_name="dpmpp_2m",
+            sampler_name="dpmpp_2m_sde",
             scheduler="karras",
-            denoise=1,
+            denoise=0.9,
             model=get_value_at_index(ipadapteradvanced_85, 0),
             positive=get_value_at_index(controlnetapplyadvanced_189, 0),
             negative=get_value_at_index(controlnetapplyadvanced_189, 1),