diff --git a/src/comfyworkflows/generate_image_flux.json b/src/comfyworkflows/generate_image_flux.json
index 012e17e..e6050dd 100644
--- a/src/comfyworkflows/generate_image_flux.json
+++ b/src/comfyworkflows/generate_image_flux.json
@@ -71,7 +71,7 @@
     },
     "38": {
         "inputs": {
-            "unet_name": "flux1-dev.safetensors",
+            "unet_name": "flux1-krea-dev.safetensors",
             "weight_dtype": "default"
         },
         "class_type": "UNETLoader",