Upload 3 files

app.py

@@ -18,7 +18,7 @@ import pandas as pd
 import numpy as np
 from pathlib import Path
 
-from env import models, models_dev, models_schnell, models_fill, models_canny, models_depth, models_edit, num_loras, num_cns, MAX_LORA, HF_TOKEN, single_file_base_models
+from env import models, models_dev, models_schnell, models_fill, models_krea, models_canny, models_depth, models_edit, num_loras, num_cns, MAX_LORA, HF_TOKEN, single_file_base_models
 from mod import (clear_cache, get_repo_safetensors, is_repo_name, is_repo_exists, get_model_trigger,
                  description_ui, compose_lora_json, is_valid_lora, fuse_loras, turbo_loras, save_image, preprocess_i2i_image,
                  get_trigger_word, enhance_prompt, set_control_union_image, get_canny_image, get_depth_image,
@@ -95,7 +95,7 @@ def load_pipeline(pipe, pipe_i2i, repo_id: str, cn_on: bool, model_type: str, ta
         controlnet_model_union_repo = 'InstantX/FLUX.1-dev-Controlnet-Union'
         if task == "Flux Fill" or repo_id in models_fill:
             model_type = "fill"
-            if repo_id in set(models_dev + models_schnell): repo_id = models_fill[0]
+            if repo_id in set(models_dev + models_schnell + models_krea): repo_id = models_fill[0]
         if dtype_str == "BF16": dtype = torch.bfloat16
         else: dtype = torch.bfloat16
         single_file_base_model = single_file_base_models.get(model_type, models[0])

env.py

@@ -65,6 +65,8 @@ models_dev = [
     "John6666/fluxescore-dev-v10fp16-fp8-flux",
     "John6666/2758-flux-asian-utopian-v30fp8noclip-fp8-flux",
     "trongg/FLUX_dev2pro_nsfw_context_lora",
+    'NikolaSigmoid/FLUX.1-Krea-dev',
+    'prithivMLmods/Flux.1-Krea-Merged-Dev',
     "https://huggingface.co/StableDiffusionVN/SDVN11-Ghibli-Flux/blob/main/SDVN11-Ghibli-Flux_fp8-hyper.safetensors",
     "https://huggingface.co/datasets/John6666/flux1-backup-202410/blob/main/iniverseMixXLSFWNSFW_f1dFP16V10.safetensors",
     "https://huggingface.co/city96/FLUX.1-dev-gguf/blob/main/flux1-dev-Q2_K.gguf",
@@ -79,13 +81,15 @@ models_fill = ["fuliucansheng/FLUX.1-Fill-dev-diffusers"]
 
 models_dedistill = []
 
+models_krea = ["NikolaSigmoid/FLUX.1-Krea-dev"]
+
 models_canny = ["fuliucansheng/FLUX.1-Canny-dev-diffusers", "sayakpaul/FLUX.1-Canny-dev-nf4"]
 
 models_depth = ["fuliucansheng/FLUX.1-Depth-dev-diffusers", "sayakpaul/FLUX.1-Depth-dev-nf4"]
 
 models_edit = ["sayakpaul/edit-control-lr_1e-4-wd_1e-4-gs_15.0-cd_0.1"]
 
-models = models_dev + models_schnell + models_fill
+models = models_dev + models_schnell + models_fill + models_krea
 
 model_trigger = {
     "Raelina/Raemu-Flux": "anime",
@@ -97,6 +101,7 @@ single_file_base_models = {
     "dev": "camenduru/FLUX.1-dev-diffusers",
     "schnell": "black-forest-labs/FLUX.1-schnell",
     "fill": "fuliucansheng/FLUX.1-Fill-dev-diffusers",
+    "krea": "NikolaSigmoid/FLUX.1-Krea-dev",
 }
 
 # List all Models for specified user

requirements.txt

@@ -1,7 +1,7 @@
 torch==2.4.0
 #torchao>=0.9.0
 #git+https://github.com/huggingface/diffusers.git
-diffusers<=0.32.0
+diffusers<=0.34.0
 transformers<=4.48.3
 #git+https://github.com/huggingface/peft.git
 peft