合并来自rayers/test的拉取请求#18

fix metadata

Author: raysers

Mflux_Comfy/Mflux_Core.py

@@ -68,7 +68,7 @@ def generate_image(prompt, model, seed, width, height, steps, guidance, quantize
     quantize = None if quantize == "None" else int(quantize)
     seed = random.randint(0, 0xffffffffffffffff) if seed == -1 else int(seed)
     print(f"Using seed: {seed}")
-    steps = min(max(steps, 2), 4) if model == "schnell" else steps
+    # steps = min(max(steps, 2), 4) if model == "schnell" else steps
 
     flux = load_or_create_flux(model, quantize, Local_model if Local_model else None, lora_paths, lora_scales, ControlNet is not None)
 
@@ -168,7 +168,7 @@ def generate_image(prompt, model, seed, width, height, steps, guidance, quantize
 
     return (tensor_image,)
 
-def save_images_with_metadata(images, prompt, model, quantize, Local_model, seed, height, width, steps, guidance, lora_paths, lora_scales, image_path, strength, filename_prefix="Mflux", full_prompt=None, extra_pnginfo=None):
+def save_images_with_metadata(images, prompt, model, quantize, Local_model, seed, height, width, steps, guidance, lora_paths, lora_scales, image_path, image_strength, filename_prefix="Mflux", full_prompt=None, extra_pnginfo=None):
 
     output_dir = folder_paths.get_output_directory()
     full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(
@@ -215,7 +215,7 @@ def save_images_with_metadata(images, prompt, model, quantize, Local_model, seed
             "guidance": guidance if model == "dev" else None,
             "Local_model": Local_model,
             "init_image_path": image_path,
-            "init_image_strength": strength,
+            "init_image_strength": image_strength,
             "lora_paths": lora_paths,
             "lora_scales": lora_scales,
         }

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "mflux-comfyui"
 description = "Simple use of [a/Mflux](https://github.com/filipstrand/mflux) in ComfyUI, suitable for users who are not familiar with terminal usage.\nNOTE: A MLX port of FLUX based on the Huggingface Diffusers implementation."
-version = "1.2.4"
+version = "1.2.5"
 license = {file = "LICENSE"}
 dependencies = ["mflux==0.4.1"]