Update app.py

app.py

@@ -20,7 +20,7 @@ import spaces
 
 token = os.environ.get("HF_TOKEN")
 
-hf_hub_download(repo_id="oimoyu/model", filename="chkp1.safetensors", local_dir="models/checkpoints")
+hf_hub_download(repo_id="oimoyu/model", filename="chkp_test.safetensors", local_dir="models/checkpoints")
 hf_hub_download(repo_id="oimoyu/model", filename="lora1.safetensors", local_dir="models/loras")
 hf_hub_download(repo_id="oimoyu/model", filename="lora2.safetensors", local_dir="models/loras")
 hf_hub_download(repo_id="oimoyu/model", filename="lora3.safetensors", local_dir="models/loras")
@@ -602,7 +602,7 @@ saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
 
 checkpointloadersimple = NODE_CLASS_MAPPINGS["CheckpointLoaderSimple"]()
 checkpointloadersimple_4 = checkpointloadersimple.load_checkpoint(
-    ckpt_name="chkp1.safetensors"
+    ckpt_name="chkp_test.safetensors"
 )
 emptylatentimage = NODE_CLASS_MAPPINGS["EmptyLatentImage"]()
 clipsetlastlayer = NODE_CLASS_MAPPINGS["CLIPSetLastLayer"]()
@@ -610,39 +610,48 @@ clipsetlastlayer_14 = clipsetlastlayer.set_last_layer(
     stop_at_clip_layer=-2, clip=get_value_at_index(checkpointloadersimple_4, 1)
 )
 
-loraloader = NODE_CLASS_MAPPINGS["LoraLoader"]()
-loraloader_11 = loraloader.load_lora(
-lora_name="lora1.safetensors",
-strength_model=0.3,
-strength_clip=0.3,
-model=get_value_at_index(checkpointloadersimple_4, 0),
-clip=get_value_at_index(clipsetlastlayer_14, 0),
-)
-
-loraloader_12 = loraloader.load_lora(
-lora_name="lora2.safetensors",
-strength_model=0.5,
-strength_clip=0.5,
-model=get_value_at_index(loraloader_11, 0),
-clip=get_value_at_index(loraloader_11, 1),
-)
-
-loraloader_12_5 = loraloader.load_lora(
-lora_name="lora4.safetensors",
-strength_model=0.25,
-strength_clip=0.25,
-model=get_value_at_index(loraloader_12, 0),
-clip=get_value_at_index(loraloader_12, 1),
-)
-
-loraloader_13 = loraloader.load_lora(
-lora_name="lora3.safetensors",
-strength_model=0.5,
-strength_clip=0.5,
-model=get_value_at_index(loraloader_12_5, 0),
-clip=get_value_at_index(loraloader_12_5, 1),
-)
-
+# loraloader = NODE_CLASS_MAPPINGS["LoraLoader"]()
+# loraloader_11 = loraloader.load_lora(
+# lora_name="lora1.safetensors",
+# strength_model=0.3,
+# strength_clip=0.3,
+# model=get_value_at_index(checkpointloadersimple_4, 0),
+# clip=get_value_at_index(clipsetlastlayer_14, 0),
+# )
+
+# loraloader_12 = loraloader.load_lora(
+# lora_name="lora2.safetensors",
+# strength_model=0.5,
+# strength_clip=0.5,
+# model=get_value_at_index(loraloader_11, 0),
+# clip=get_value_at_index(loraloader_11, 1),
+# )
+
+# loraloader_12_5 = loraloader.load_lora(
+# lora_name="lora4.safetensors",
+# strength_model=0.25,
+# strength_clip=0.25,
+# model=get_value_at_index(loraloader_12, 0),
+# clip=get_value_at_index(loraloader_12, 1),
+# )
+
+# loraloader_13 = loraloader.load_lora(
+# lora_name="lora3.safetensors",
+# strength_model=0.5,
+# strength_clip=0.5,
+# model=get_value_at_index(loraloader_12_5, 0),
+# clip=get_value_at_index(loraloader_12_5, 1),
+# )
+
+
+# applyfbcacheonmodel_16 = applyfbcacheonmodel.patch(
+#     object_to_patch="diffusion_model",
+#     residual_diff_threshold=0.2,
+#     start=0.7,
+#     end=1,
+#     max_consecutive_cache_hits=-1,
+#     model=get_value_at_index(loraloader_13, 0),
+# )
 
 applyfbcacheonmodel_16 = applyfbcacheonmodel.patch(
     object_to_patch="diffusion_model",
@@ -650,7 +659,7 @@ applyfbcacheonmodel_16 = applyfbcacheonmodel.patch(
     start=0.7,
     end=1,
     max_consecutive_cache_hits=-1,
-    model=get_value_at_index(loraloader_13, 0),
+    model=get_value_at_index(checkpointloadersimple_4, 0),
 )
 
 wd14taggerpysssss = NODE_CLASS_MAPPINGS["WD14Tagger|pysssss"]()
@@ -665,8 +674,10 @@ model_management.load_models_gpu([
 ])
 
 
-LOADED_MODEL = get_value_at_index(loraloader_13, 0)
-LOADED_CLIP = get_value_at_index(loraloader_13, 1) 
+# LOADED_MODEL = get_value_at_index(loraloader_13, 0)
+# LOADED_CLIP = get_value_at_index(loraloader_13, 1) 
+LOADED_MODEL = get_value_at_index(checkpointloadersimple_4, 0)
+LOADED_CLIP = get_value_at_index(checkpointloadersimple_4, 1) 
 LOADED_VAE = get_value_at_index(checkpointloadersimple_4, 2)
 LOADED_WAVESPEED_MODEL = get_value_at_index(applyfbcacheonmodel_16, 0)
     
@@ -736,7 +747,7 @@ if __name__ == "__main__":
                 
                 with gr.Column():
                     # 输出图像
-                    output_image = gr.Image(label="Generated Image", show_label=False, format="JPEG")
+                    output_image = gr.Image(label="Generated Image", show_label=False, format="png")
                 
                 # 当点击按钮时，它将触发"generate_image"函数，该函数带有相应的输入
                 # 并且输出是一张图像
@@ -800,7 +811,7 @@ if __name__ == "__main__":
                 
                 with gr.Column():
                     # 输出图像
-                    i2i_output_image = gr.Image(label="Generated Image", show_label=False, format="JPEG")
+                    i2i_output_image = gr.Image(label="Generated Image", show_label=False, format="png")
                 
                 i2i_generate_btn.click(
                     fn=infer_i2i,