Upload 4 files

app.py

@@ -384,6 +384,35 @@ with gr.Blocks(fill_width=True, elem_id="container", css=css, delete_cache=(60,
         copy_prompt_btn.click(gradio_copy_prompt, inputs=[output_text], outputs=[prompt], show_api=False)
         copy_prompt_btn_pony.click(gradio_copy_prompt, inputs=[output_text_pony], outputs=[prompt], show_api=False)
 
+    with gr.Tab("PNG Info"):
+        def extract_exif_data(image):
+            if image is None: return ""
+
+            try:
+                metadata_keys = ['parameters', 'metadata', 'prompt', 'Comment']
+
+                for key in metadata_keys:
+                    if key in image.info:
+                        return image.info[key]
+
+                return str(image.info)
+
+            except Exception as e:
+                return f"Error extracting metadata: {str(e)}"
+
+        with gr.Row():
+            with gr.Column():
+                image_metadata = gr.Image(label="Image with metadata", type="pil", sources=["upload"])
+
+            with gr.Column():
+                result_metadata = gr.Textbox(label="Metadata", show_label=True, show_copy_button=True, interactive=False, container=True, max_lines=99)
+
+                image_metadata.change(
+                    fn=extract_exif_data,
+                    inputs=[image_metadata],
+                    outputs=[result_metadata],
+                )
+
     gr.LoginButton()
     gr.DuplicateButton(value="Duplicate Space for private use (This demo does not work on CPU. Requires GPU Space)")
 

dc.py

@@ -23,6 +23,9 @@ from stablepy import (
 )
 import time
 #import urllib.parse
+
+print(os.getenv("SPACES_ZERO_GPU"))
+
 import gradio as gr
 import logging
 logging.getLogger("diffusers").setLevel(logging.ERROR)
@@ -52,14 +55,14 @@ PREPROCESSOR_CONTROLNET = {
   ],
   "scribble": [
     "HED",
-    "Pidinet",
+    "PidiNet",
     "None",
   ],
   "softedge": [
-    "Pidinet",
+    "PidiNet",
     "HED",
     "HED safe",
-    "Pidinet safe",
+    "PidiNet safe",
     "None",
   ],
   "segmentation": [
@@ -295,18 +298,21 @@ def extract_parameters(input_string):
     input_string = input_string.replace("\n", "")
 
     if "Negative prompt:" not in input_string:
-        print("Negative prompt not detected")
-        parameters["prompt"] = input_string
-        return parameters
+        if "Steps:" in input_string:
+            input_string = input_string.replace("Steps:", "Negative prompt: Steps:")
+        else:
+            print("Invalid metadata")
+            parameters["prompt"] = input_string
+            return parameters
 
     parm = input_string.split("Negative prompt:")
-    parameters["prompt"] = parm[0]
+    parameters["prompt"] = parm[0].strip()
     if "Steps:" not in parm[1]:
         print("Steps not detected")
-        parameters["neg_prompt"] = parm[1]
+        parameters["neg_prompt"] = parm[1].strip()
         return parameters
     parm = parm[1].split("Steps:")
-    parameters["neg_prompt"] = parm[0]
+    parameters["neg_prompt"] = parm[0].strip()
     input_string = "Steps:" + parm[1]
 
     # Extracting Steps
@@ -355,12 +361,13 @@ class GuiSD:
             retain_task_model_in_cache=False,
             device="cpu",
         )
-        self.model.device = torch.device("cpu") #
+        self.model.load_beta_styles()
+        #self.model.device = torch.device("cpu") #
 
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):
-        progress(0, desc="Start inference...")
+        #progress(0, desc="Start inference...")
         images, seed, image_list, metadata = model(**pipe_params)
-        progress(1, desc="Inference completed.")
+        #progress(1, desc="Inference completed.")
         if not isinstance(images, list): images = [images]
         images = save_images(images, metadata)
         img = []
@@ -500,7 +507,7 @@ class GuiSD:
         pag_scale,
         progress=gr.Progress(track_tqdm=True),
     ):
-        progress(0, desc="Preparing inference...")
+        #progress(0, desc="Preparing inference...")
         
         vae_model = vae_model if vae_model != "None" else None
         loras_list = [lora1, lora2, lora3, lora4, lora5]
@@ -670,7 +677,7 @@ class GuiSD:
             self.model.pipe.transformer.to(self.model.device)
             print("transformer to cuda")
 
-        progress(1, desc="Inference preparation completed. Starting inference...")
+        #progress(1, desc="Inference preparation completed. Starting inference...")
 
         info_state = "" # for yield version
         return self.infer_short(self.model, pipe_params, progress), info_state
@@ -805,9 +812,12 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     lora3 = get_valid_lora_path(lora3)
     lora4 = get_valid_lora_path(lora4)
     lora5 = get_valid_lora_path(lora5)
-    progress(1, desc="Preparation completed. Starting inference preparation...")
+    progress(1, desc="Preparation completed. Starting inference...")
 
-    sd_gen.load_new_model(model_name, vae, TASK_MODEL_LIST[0], progress)
+    progress(0, desc="Loading model...")
+    sd_gen.load_new_model(model_name, vae, TASK_MODEL_LIST[0])
+    progress(1, desc="Model loaded.")
+    progress(0, desc="Starting Inference...")
     images, info = sd_gen_generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
         guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
         lora4, lora4_wt, lora5, lora5_wt, sampler,
@@ -820,7 +830,6 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         True, None, None, "plus_face", "original", 0.7, None, None, "base", "style", 0.7, 0.0,
         load_lora_cpu, verbose_info, gpu_duration
     )
-
     progress(1, desc="Inference completed.")
     output_image = images[0][0] if images else None
 

env.py

@@ -39,6 +39,11 @@ load_diffusers_format_model = [
     'votepurchase/juggernautXL_hyper_8step_sfw',
     'votepurchase/ponyRealism_v21MainVAE',
     'stabilityai/stable-diffusion-xl-base-1.0',
+    'John6666/blue-pencil-flux1-v021-fp8-flux',
+    'John6666/wai-ani-flux-v10forfp8-fp8-flux',
+    'John6666/xe-anime-flux-v04-fp8-flux',
+    'John6666/lyh-anime-flux-v2a1-fp8-flux',
+    'John6666/carnival-unchained-v10-fp8-flux',
     'cagliostrolab/animagine-xl-3.1',
     'John6666/epicrealism-xl-v8kiss-sdxl',
     'misri/epicrealismXL_v7FinalDestination',
@@ -54,15 +59,30 @@ load_diffusers_format_model = [
     'WhiteAiZ/autismmixSDXL_autismmixConfetti_diffusers',
     'kitty7779/ponyDiffusionV6XL',
     'GraydientPlatformAPI/aniverse-pony',
+    'John6666/ras-real-anime-screencap-v1-sdxl',
+    'John6666/duchaiten-pony-xl-no-score-v60-sdxl',
     'John6666/mistoon-anime-ponyalpha-sdxl',
+    'John6666/3x3x3mixxl-v2-sdxl',
+    'John6666/3x3x3mixxl-3dv01-sdxl',
     'John6666/ebara-mfcg-pony-mix-v12-sdxl',
     'John6666/t-ponynai3-v51-sdxl',
+    'John6666/t-ponynai3-v65-sdxl',
+    'John6666/prefect-pony-xl-v3-sdxl',
     'John6666/mala-anime-mix-nsfw-pony-xl-v5-sdxl',
     'John6666/wai-real-mix-v11-sdxl',
+    'John6666/wai-c-v6-sdxl',
+    'John6666/iniverse-mix-xl-sfwnsfw-pony-guofeng-v43-sdxl',
+    'John6666/photo-realistic-pony-v5-sdxl',
+    'John6666/pony-realism-v21main-sdxl',
+    'John6666/pony-realism-v22main-sdxl',
     'John6666/cyberrealistic-pony-v63-sdxl',
+    'John6666/cyberrealistic-pony-v64-sdxl',
     'GraydientPlatformAPI/realcartoon-pony-diffusion',
     'John6666/nova-anime-xl-pony-v5-sdxl',
     'John6666/autismmix-sdxl-autismmix-pony-sdxl',
+    'John6666/aimz-dream-real-pony-mix-v3-sdxl',
+    'John6666/duchaiten-pony-real-v11fix-sdxl',
+    'John6666/duchaiten-pony-real-v20-sdxl',
     'yodayo-ai/kivotos-xl-2.0',
     'yodayo-ai/holodayo-xl-2.1',
     'yodayo-ai/clandestine-xl-1.0',
@@ -81,6 +101,8 @@ load_diffusers_format_model = [
     'GraydientPlatformAPI/picx-real',
     'GraydientPlatformAPI/perfectworld6',
     'emilianJR/epiCRealism',
+    'votepurchase/counterfeitV30_v30',
+    'votepurchase/ChilloutMix',
     'Meina/MeinaMix_V11',
     'Meina/MeinaUnreal_V5',
     'Meina/MeinaPastel_V7',

llmdolphin.py

@@ -20,6 +20,7 @@ llm_models = {
     #"": ["", MessagesFormatterType.PHI_3],
     "mn-12b-lyra-v2a1-q5_k_m.gguf": ["HalleyStarbun/MN-12B-Lyra-v2a1-Q5_K_M-GGUF", MessagesFormatterType.CHATML],
     "L3-8B-Tamamo-v1.i1-Q5_K_M.gguf": ["mradermacher/L3-8B-Tamamo-v1-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Mahou-1.5-mistral-nemo-12B.i1-Q4_K_M.gguf": ["mradermacher/Mahou-1.5-mistral-nemo-12B-i1-GGUF", MessagesFormatterType.MISTRAL],
     "MN-12B-Mag-Mell-Q4_K_M.gguf": ["inflatebot/MN-12B-Mag-Mell-R1-GGUF", MessagesFormatterType.MISTRAL],
     "Qwen2.5-14B-Instruct-Q4_K_M.gguf": ["bartowski/Qwen2.5-14B-Instruct-GGUF", MessagesFormatterType.OPEN_CHAT],
     "Instant-RP-Noodles-12B-v1.3.Q4_K_M.gguf": ["mradermacher/Instant-RP-Noodles-12B-v1.3-GGUF", MessagesFormatterType.MISTRAL],
@@ -62,6 +63,30 @@ llm_models = {
     "ChatWaifu_22B_v2.0_preview.Q4_K_S.gguf": ["mradermacher/ChatWaifu_22B_v2.0_preview-GGUF", MessagesFormatterType.MISTRAL],
     "ChatWaifu_v1.4.Q5_K_M.gguf": ["mradermacher/ChatWaifu_v1.4-GGUF", MessagesFormatterType.MISTRAL],
     "ChatWaifu_v1.3.1.Q4_K_M.gguf": ["mradermacher/ChatWaifu_v1.3.1-GGUF", MessagesFormatterType.MISTRAL],
+    "Humanish-Qwen2.5-7B-Instruct-Q5_K_M.gguf": ["bartowski/Humanish-Qwen2.5-7B-Instruct-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "FactAlign-LLaMA-3-8B.i1-Q4_K_M.gguf": ["mradermacher/FactAlign-LLaMA-3-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Magot-v3-Gemma2-8k-9B.i1-Q4_K_M.gguf": ["mradermacher/Magot-v3-Gemma2-8k-9B-i1-GGUF", MessagesFormatterType.ALPACA],
+    "EVA-Qwen2.5-14B-v0.1-Q4_K_M.gguf": ["bartowski/EVA-Qwen2.5-14B-v0.1-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Flammades-Mistral-Nemo-12B.i1-Q4_K_M.gguf": ["mradermacher/Flammades-Mistral-Nemo-12B-i1-GGUF", MessagesFormatterType.CHATML],
+    "Humanish-LLama3-8B-Instruct-Q5_K_M.gguf": ["bartowski/Humanish-LLama3-8B-Instruct-GGUF", MessagesFormatterType.LLAMA_3],
+    "Humanish-Mistral-Nemo-Instruct-2407-Q4_K_M.gguf": ["bartowski/Humanish-Mistral-Nemo-Instruct-2407-GGUF", MessagesFormatterType.MISTRAL],
+    "L3.1-Clouded-Uchtave-v0.1-8B.Q5_K_M.gguf": ["mradermacher/L3.1-Clouded-Uchtave-v0.1-8B-GGUF", MessagesFormatterType.LLAMA_3],
+    "Baldur-Q5_K_M.gguf": ["Delta-Vector/Baldur-8B-GGUF", MessagesFormatterType.LLAMA_3],
+    "llama-jpsft-q6_k.gguf": ["ai-net/Llama-JPSFT-GGUF", MessagesFormatterType.LLAMA_3],
+    "Gemma-2-Ataraxy-v3i-9B.Q5_K_M.gguf": ["mradermacher/Gemma-2-Ataraxy-v3i-9B-GGUF", MessagesFormatterType.ALPACA],
+    "EVA-Qwen2.5-7B-v0.1.i1-Q5_K_M.gguf": ["mradermacher/EVA-Qwen2.5-7B-v0.1-i1-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "EZO-Llama-3.2-3B-Instruct-dpoE.Q4_K_S.gguf": ["mradermacher/EZO-Llama-3.2-3B-Instruct-dpoE-GGUF", MessagesFormatterType.LLAMA_3],
+    "Bezaliel_p1.1-12B.i1-Q4_K_M.gguf": ["mradermacher/Bezaliel_p1.1-12B-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "Ice0.7-29.09-RP.i1-Q5_K_M.gguf": ["mradermacher/Ice0.7-29.09-RP-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "IceWhiskeyRP-7b.i1-Q5_K_M.gguf": ["mradermacher/IceWhiskeyRP-7b-i1-GGUF", MessagesFormatterType.ALPACA],
+    "L3.1-Purosani-2-8B.i1-Q4_K_M.gguf": ["mradermacher/L3.1-Purosani-2-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "llama-3.2-3b-instruct-abliterated-q4_k_m.gguf": ["darkc0de/Llama-3.2-3B-Instruct-abliterated-Q4_K_M-GGUF", MessagesFormatterType.LLAMA_3],
+    "Daredevil-Aura-8B_uncensored_OAS_abliterated.i1-Q4_K_S.gguf": ["mradermacher/Daredevil-Aura-8B_uncensored_OAS_abliterated-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Qwen-2.5-7b-TitanFusion-v3.Q5_K_M.gguf": ["mradermacher/Qwen-2.5-7b-TitanFusion-v3-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "MixTAO-7Bx2-MoE-v8.1.Q4_K_M.gguf": ["mradermacher/MixTAO-7Bx2-MoE-v8.1-GGUF", MessagesFormatterType.ALPACA],
+    "Promissum_Mane-8B-LINEAR.i1-Q5_K_M.gguf": ["mradermacher/Promissum_Mane-8B-LINEAR-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "L3.1-Purosani-1.5-8B.Q5_K_M.gguf": ["mradermacher/L3.1-Purosani-1.5-8B-GGUF", MessagesFormatterType.LLAMA_3],
+    "SzilviaB-Daredevil-LongWriter-8B_abliterated.i1-Q5_K_M.gguf": ["mradermacher/SzilviaB-Daredevil-LongWriter-8B_abliterated-i1-GGUF", MessagesFormatterType.LLAMA_3],
     "eidolon-v1-14b-q4_k_m.gguf": ["Lambent/Eidolon-v1-14B-Q4_K_M-GGUF", MessagesFormatterType.OPEN_CHAT],
     "Mistral-Nemo-Gutenberg-Doppel-12B-v2.Q4_K_M.gguf": ["QuantFactory/Mistral-Nemo-Gutenberg-Doppel-12B-v2-GGUF", MessagesFormatterType.MISTRAL],
     "MN-Dark-Planet-TITAN-12B-D_AU-Q4_k_s.gguf": ["DavidAU/MN-Dark-Planet-TITAN-12B-GGUF", MessagesFormatterType.CHATML],