Upload 2 files

mod.py

@@ -24,7 +24,9 @@ models = [
     "John6666/copycat-flux-test-fp8-v11-fp8-flux",
     "John6666/nepotism-fuxdevschnell-v3aio-fp8-flux",
     "John6666/niji-style-flux-devfp8-fp8-flux",
+    "John6666/niji56-style-v3-fp8-flux",
     "John6666/lyh-dalle-anime-v12dalle-fp8-flux",
+    "John6666/xe-hentai-flux-01-fp8-flux",
     "John6666/fluxunchained-artfulnsfw-fut516xfp8e4m3fnv11-fp8-flux",
     "John6666/fastflux-unchained-t5f16-fp8-flux",
     "John6666/iniverse-mix-xl-sfwnsfw-fluxdfp8v10-fp8-flux",
@@ -164,12 +166,11 @@ def get_control_params():
 
 from preprocessor import Preprocessor
 def preprocess_image(image: Image.Image, control_mode: str, height: int, width: int,
-                     preprocess_resolution: int, progress=gr.Progress(track_tqdm=True)):
+                     preprocess_resolution: int):
     if control_mode == "None": return image
     image_resolution = max(width, height)
     image_before = resize_image(expand2square(image.convert("RGB")), image_resolution, image_resolution, False)
     # generated control_
-    progress(0, desc="start to generate control image")
     print("start to generate control image")
     preprocessor = Preprocessor()
     if control_mode == "depth_midas":
@@ -221,7 +222,6 @@ def preprocess_image(image: Image.Image, control_mode: str, height: int, width:
     
     image_after = resize_image(control_image, width, height, False)
     ref_width, ref_height = image.size
-    progress(1, desc=f"generate control image success: {ref_width}x{ref_height} => {image_width}x{image_height}")
     print(f"generate control image success: {ref_width}x{ref_height} => {image_width}x{image_height}")
     return image_after
 
@@ -271,9 +271,8 @@ def get_trigger_word(lorajson: list[dict]):
 
 # https://huggingface.co/docs/diffusers/v0.23.1/en/api/loaders#diffusers.loaders.LoraLoaderMixin.fuse_lora
 # https://github.com/huggingface/diffusers/issues/4919
-def fuse_loras(pipe, lorajson: list[dict], progress=gr.Progress(track_tqdm=True)):
+def fuse_loras(pipe, lorajson: list[dict]):
     if not lorajson or not isinstance(lorajson, list): return
-    progress(0, desc="Fusing LoRA.")
     a_list = []
     w_list = []
     for d in lorajson: