RealVis Refiner based on V5

app.py

@@ -1,352 +1,82 @@
-#!/usr/bin/env python
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-
-import os
-import random
-import uuid
-import gradio as gr
-import numpy as np
+from diffusers import EulerAncestralDiscreteScheduler, StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline
 from PIL import Image
-import spaces
+import gradio as gr
+import random
 import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
-from typing import Tuple
+import math
 
+device = "cuda"
+negative_prompt = "deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime, text, close up, cropped, out of frame, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck"
 
-css = '''
-.gradio-container{max-width: 575px !important}
-h1{text-align:center}
-footer {
-    visibility: hidden
+css = """
+.btn-green {
+  background-image: linear-gradient(to bottom right, #6dd178, #00a613) !important;
+  border-color: #22c55e !important;
+  color: #166534 !important;
 }
-'''
-
-
-DESCRIPTIONXX = """
-    ## REALVISXL V5.0 ⚡
-"""
-
-examples = [
-
-    "Many apples splashed with drops of water within a fancy bowl 4k, hdr  --v 6.0 --style raw",
-    "A profile photo of a dog, brown background, shot on Leica M6 --ar 128:85 --v 6.0 --style raw",
-]
-
-MODEL_OPTIONS = {
-    "REALVISXL V5.0": "SG161222/RealVisXL_V5.0",
-    "LIGHTNING V5.0": "SG161222/RealVisXL_V5.0_Lightning",
+.btn-green:hover {
+  background-image: linear-gradient(to bottom right, #6dd178, #6dd178) !important;
 }
+"""
 
-MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
-USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
-ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
-BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
-
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
-style_list = [
-    {
-        "name": "3840 x 2160",
-        "prompt": "hyper-realistic 8K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-    {
-        "name": "2560 x 1440",
-        "prompt": "hyper-realistic 4K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-    {
-        "name": "HD+",
-        "prompt": "hyper-realistic 2K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-    {
-        "name": "Style Zero",
-        "prompt": "{prompt}",
-        "negative_prompt": "",
-    },
-]
-
-styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
-DEFAULT_STYLE_NAME = "3840 x 2160"
-STYLE_NAMES = list(styles.keys())
-
-def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
-    if style_name in styles:
-        p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
-    else:
-        p, n = styles[DEFAULT_STYLE_NAME]
-
-    if not negative:
-        negative = ""
-    return p.replace("{prompt}", positive), n + negative
-
-def load_and_prepare_model(model_id):
-    pipe = StableDiffusionXLPipeline.from_pretrained(
-        model_id,
+def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
+    print("prompt = ", prompt)
+    turbo_steps=5 # use sweet spot
+    if seed < 0:
+        seed = random.randint(1,999999)
+    image = txt2img(
+        prompt=prompt,
+		negative_prompt=negative_prompt,
+        num_inference_steps=turbo_steps,  
+        guidance_scale=2,
+        seed=seed,
+    ).images[0]
+    upscaled_image = image.resize((1024,1024), 1)
+    final_image = img2img(
+        prompt=prompt,
+		negative_prompt=negative_prompt,
+        image=upscaled_image,
+        num_inference_steps=max(1,samp_steps-turbo_steps*2), # always discount lightning at 2x efficiency
+        guidance_scale=5,
+        strength=1,
+        seed=seed,
+    ).images[0]
+    return [final_image], seed
+
+
+def set_base_models():
+    txt2img = StableDiffusionXLPipeline.from_pretrained(
+        "SG161222/RealVisXL_V5.0_Lightning",
         torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
         use_safetensors=True,
         add_watermarker=False,
     ).to(device)
-    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    
-    if USE_TORCH_COMPILE:
-        pipe.compile()
-    
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    
-    return pipe
-
-# Preload and compile both models
-models = {key: load_and_prepare_model(value) for key, value in MODEL_OPTIONS.items()}
-
-MAX_SEED = np.iinfo(np.int32).max
-
-def save_image(img):
-    unique_name = str(uuid.uuid4()) + ".png"
-    img.save(unique_name)
-    return unique_name
-
-def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    return seed
-
-@spaces.GPU(duration=60, enable_queue=True)
-def generate(
-    model_choice: str,
-    prompt: str,
-    negative_prompt: str = "",
-    use_negative_prompt: bool = False,
-    style_selection: str = DEFAULT_STYLE_NAME,
-    seed: int = 1,
-    width: int = 1024,
-    height: int = 1024,
-    guidance_scale: float = 3,
-    num_inference_steps: int = 25,
-    randomize_seed: bool = False,
-    use_resolution_binning: bool = True, 
-    num_images: int = 1,  
-    progress=gr.Progress(track_tqdm=True),
-):
-    global models
-    pipe = models[model_choice]
-    
-    seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device=device).manual_seed(seed)
-
-    prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
-
-    options = {
-        "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
-        "width": width,
-        "height": height,
-        "guidance_scale": guidance_scale,
-        "num_inference_steps": num_inference_steps,
-        "generator": generator,
-        "output_type": "pil",
-    }
-
-    if use_resolution_binning:
-        options["use_resolution_binning"] = True
-
-    images = []
-    for i in range(0, num_images, BATCH_SIZE):
-        batch_options = options.copy()
-        batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
-        if "negative_prompt" in batch_options:
-            batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
-        images.extend(pipe(**batch_options).images)
-
-    image_paths = [save_image(img) for img in images]
-    return image_paths, seed
-
-def load_predefined_images1():
-    predefined_images1 = [
-        "assets/7.png",
-        "assets/8.png",
-        "assets/9.png",
-        "assets/1.png",
-        "assets/2.png",
-        "assets/3.png",
-        "assets/4.png",
-        "assets/5.png",
-        "assets/6.png",
-    ]
-    return predefined_images1
-
-
-def load_predefined_images():
-     predefined_images = [
-         "assets2/11.png",
-         "assets2/22.png",
-         "assets2/33.png",
-         "assets2/44.png",
-         "assets2/55.png",
-         "assets2/66.png",
-         "assets2/77.png",
-         "assets2/88.png",
-         "assets2/99.png",
-     ]
-     return predefined_images
-
-
-with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
-    gr.Markdown(DESCRIPTIONXX)
-    with gr.Row():
-        prompt = gr.Text(
-            label="Prompt",
-            show_label=False,
-            max_lines=1,
-            placeholder="Enter your prompt",
-            container=False,
-        )
-        run_button = gr.Button("Run", scale=0)
-    result = gr.Gallery(label="Result", columns=1, show_label=False) 
+    txt2img.scheduler = EulerAncestralDiscreteScheduler.from_config(txt2img.scheduler.config)
+	
+    img2img = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+        "SG161222/RealVisXL_V5.0",
+        torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32,
+		use_safetensors=True,
+        add_watermarker=False,
+    ).to(device)
+    img2img.scheduler = EulerAncestralDiscreteScheduler.from_config(img2img.scheduler.config)
 
-    with gr.Row():
-        model_choice = gr.Dropdown(
-            label="Model Selection🔻",
-            choices=list(MODEL_OPTIONS.keys()),
-            value="LIGHTNING V5.0"
-        )
+    return txt2img, img2img
 
-    with gr.Accordion("Advanced options", open=False, visible=False):
-        style_selection = gr.Radio(
-            show_label=True,
-            container=True,
-            interactive=True,
-            choices=STYLE_NAMES,
-            value=DEFAULT_STYLE_NAME,
-            label="Quality Style",
-        )
-        num_images = gr.Slider(
-            label="Number of Images",
-            minimum=1,
-            maximum=5,
-            step=1,
-            value=1,
-        )
-        with gr.Row():
-            with gr.Column(scale=1):
-                use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
-                negative_prompt = gr.Text(
-                    label="Negative prompt",
-                    max_lines=5,
-                    lines=4,
-                    placeholder="Enter a negative prompt",
-                    value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
-                    visible=True,
-                )
-        seed = gr.Slider(
-            label="Seed",
-            minimum=0,
-            maximum=MAX_SEED,
-            step=1,
-            value=0,
-        )
-        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+with gr.Blocks(css=css) as demo:
+    with gr.Column():
+        prompt = gr.Textbox(label="Prompt for RealVisXL_V5")
+        submit_btn = gr.Button("Generate", elem_classes="btn-green")
+        
         with gr.Row():
-            width = gr.Slider(
-                label="Width",
-                minimum=512,
-                maximum=MAX_IMAGE_SIZE,
-                step=64,
-                value=1024,
-            )
-            height = gr.Slider(
-                label="Height",
-                minimum=512,
-                maximum=MAX_IMAGE_SIZE,
-                step=64,
-                value=1024,
-            )
-        with gr.Row():
-            guidance_scale = gr.Slider(
-                label="Guidance Scale",
-                minimum=0.1,
-                maximum=6,
-                step=0.1,
-                value=3.0,
-            )
-            num_inference_steps = gr.Slider(
-                label="Number of inference steps",
-                minimum=1,
-                maximum=60,
-                step=1,
-                value=28,
-            )
-
-    gr.Examples(
-        examples=examples,
-        inputs=prompt,
-        cache_examples=False
-    )
-
-    use_negative_prompt.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_negative_prompt,
-        outputs=negative_prompt,
-        api_name=False,
-    )
-    
-    gr.on(
-        triggers=[
-            prompt.submit,
-            negative_prompt.submit,
-            run_button.click,
-        ],
-        fn=generate,
-        inputs=[
-            model_choice,
-            prompt,
-            negative_prompt,
-            use_negative_prompt,
-            style_selection,
-            seed,
-            width,
-            height,
-            guidance_scale,
-            num_inference_steps,
-            randomize_seed, 
-            num_images, 
-        ],
-        outputs=[result, seed],
-    )
-
-    gr.Markdown("### LIGHTNING V5.0")
-    predefined_gallery = gr.Gallery(label="LIGHTNING V5.0", columns=3, show_label=False, value=load_predefined_images())
-    
-    gr.Markdown("### REALVISXL V5.0")
-    predefined_gallery = gr.Gallery(label="REALVISXL V5.0", columns=3, show_label=False, value=load_predefined_images1())
-
-    gr.Markdown(
-    """
-    <div style="text-align: justify;">
-    ⚡Models used in the playground <a href="https://huggingface.co/SG161222/RealVisXL_V5.0">[REALVISXL V5.0]</a>, <a href="https://huggingface.co/SG161222/RealVisXL_V5.0_Lightning">[REALVISXL V5.0 LIGHTNING]</a> for image generation. Stable Diffusion XL piped (SDXL) model HF. This is the demo space for generating images using the Stable Diffusion XL models, with multiple different variants available.
-    </div>
-    """)
-
-    gr.Markdown(
-    """
-    <div style="text-align: justify;">
-    ⚡This is the demo space for generating images using Stable Diffusion XL with quality styles, different models, and types. Try the sample prompts to generate higher quality images. Try the sample prompts for generating higher quality images. 
-    <a href='https://huggingface.co/spaces/prithivMLmods/Top-Prompt-Collection' target='_blank'>Try prompts</a>.
-    </div>
-    """)
-
-    gr.Markdown(
-    """
-    <div style="text-align: justify;">
-    ⚠️ Users are accountable for the content they generate and are responsible for ensuring it meets appropriate ethical standards.
-    </div>
-    """) 
+            turbo_steps = gr.Slider(5, 5, value=5, step=1, label="Turbo steps (fixed at 5)")
+            sampling_steps = gr.Slider(15, 50, value=25, step=1, label="Effective steps")
+            seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
+            lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
+            
+        gallery = gr.Gallery(show_label=False, preview=True, container=False, height=1100)
+        
+    submit_btn.click(generate, [prompt, sampling_steps, seed], [gallery, lastSeed], queue=True)
     
-if __name__ == "__main__":
-    demo.queue(max_size=50).launch(show_api=False)
+txt2img, img2img = set_base_models()
+demo.launch(debug=True, share=True)

files/demo.txt

@@ -1,285 +0,0 @@
-#!/usr/bin/env python
-import os
-import random
-import uuid
-import gradio as gr
-import numpy as np
-from PIL import Image
-import spaces
-import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
-from typing import Tuple
-
-css = '''
-.gradio-container{max-width: 570px !important}
-h1{text-align:center}
-footer {
-    visibility: hidden
-}
-'''
-
-DESCRIPTIONXX = """
-    ## REALVISXL V5 + LIGHTNING ⚡
-"""
-
-examples = [
-    "Illustration of A starry night camp in the mountains, 4k, cinematic --ar 85:128 --v 6.0 --style raw",
-    "A delicious ceviche cheesecake slice, 4k, octane render, ray tracing, Ultra-High-Definition"
-]
-
-MODEL_OPTIONS = {
-    "REALVISXL V5.0": "SG161222/RealVisXL_V5.0",
-    # "LIGHTNING V5.0": "SG161222/RealVisXL_V5.0_Lightning",
-}
-
-MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
-USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
-ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
-BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
-
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
-style_list = [
-    {
-        "name": "3840 x 2160",
-        "prompt": "hyper-realistic 8K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-    {
-        "name": "2560 x 1440",
-        "prompt": "hyper-realistic 4K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-    {
-        "name": "HD+",
-        "prompt": "hyper-realistic 2K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-    {
-        "name": "Style Zero",
-        "prompt": "{prompt}",
-        "negative_prompt": "",
-    },
-]
-
-styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
-DEFAULT_STYLE_NAME = "3840 x 2160"
-STYLE_NAMES = list(styles.keys())
-
-def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
-    if style_name in styles:
-        p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
-    else:
-        p, n = styles[DEFAULT_STYLE_NAME]
-
-    if not negative:
-        negative = ""
-    return p.replace("{prompt}", positive), n + negative
-
-def load_and_prepare_model(model_id):
-    pipe = StableDiffusionXLPipeline.from_pretrained(
-        model_id,
-        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-        use_safetensors=True,
-        add_watermarker=False,
-    ).to(device)
-    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    
-    if USE_TORCH_COMPILE:
-        pipe.compile()
-    
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    
-    return pipe
-
-# Preload and compile both models
-models = {key: load_and_prepare_model(value) for key, value in MODEL_OPTIONS.items()}
-
-MAX_SEED = np.iinfo(np.int32).max
-
-def save_image(img):
-    unique_name = str(uuid.uuid4()) + ".png"
-    img.save(unique_name)
-    return unique_name
-
-def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    return seed
-
-@spaces.GPU(duration=60, enable_queue=True)
-def generate(
-    model_choice: str,
-    prompt: str,
-    negative_prompt: str = "",
-    use_negative_prompt: bool = False,
-    style_selection: str = DEFAULT_STYLE_NAME,
-    seed: int = 1,
-    width: int = 1024,
-    height: int = 1024,
-    guidance_scale: float = 3,
-    num_inference_steps: int = 25,
-    randomize_seed: bool = False,
-    use_resolution_binning: bool = True, 
-    num_images: int = 1,  
-    progress=gr.Progress(track_tqdm=True),
-):
-    global models
-    pipe = models[model_choice]
-    
-    seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device=device).manual_seed(seed)
-
-    prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
-
-    options = {
-        "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
-        "width": width,
-        "height": height,
-        "guidance_scale": guidance_scale,
-        "num_inference_steps": num_inference_steps,
-        "generator": generator,
-        "output_type": "pil",
-    }
-
-    if use_resolution_binning:
-        options["use_resolution_binning"] = True
-
-    images = []
-    for i in range(0, num_images, BATCH_SIZE):
-        batch_options = options.copy()
-        batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
-        if "negative_prompt" in batch_options:
-            batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
-        images.extend(pipe(**batch_options).images)
-
-    image_paths = [save_image(img) for img in images]
-    return image_paths, seed
-
-with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
-    gr.Markdown(DESCRIPTIONXX)
-    with gr.Row():
-        prompt = gr.Text(
-            label="Prompt",
-            show_label=False,
-            max_lines=1,
-            placeholder="Enter your prompt",
-            container=False,
-        )
-        run_button = gr.Button("Run", scale=0)
-    result = gr.Gallery(label="Result", columns=1, show_label=False) 
-
-    with gr.Row():
-        model_choice = gr.Dropdown(
-            label="Model Selection🔻",
-            choices=list(MODEL_OPTIONS.keys()),
-            value="REALVISXL V5.0"
-        )
-
-    with gr.Accordion("Advanced options", open=False, visible=True):
-        style_selection = gr.Radio(
-            show_label=True,
-            container=True,
-            interactive=True,
-            choices=STYLE_NAMES,
-            value=DEFAULT_STYLE_NAME,
-            label="Quality Style",
-        )
-        num_images = gr.Slider(
-            label="Number of Images",
-            minimum=1,
-            maximum=5,
-            step=1,
-            value=1,
-        )
-        with gr.Row():
-            with gr.Column(scale=1):
-                use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
-                negative_prompt = gr.Text(
-                    label="Negative prompt",
-                    max_lines=5,
-                    lines=4,
-                    placeholder="Enter a negative prompt",
-                    value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
-                    visible=True,
-                )
-        seed = gr.Slider(
-            label="Seed",
-            minimum=0,
-            maximum=MAX_SEED,
-            step=1,
-            value=0,
-        )
-        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-        with gr.Row():
-            width = gr.Slider(
-                label="Width",
-                minimum=512,
-                maximum=MAX_IMAGE_SIZE,
-                step=64,
-                value=1024,
-            )
-            height = gr.Slider(
-                label="Height",
-                minimum=512,
-                maximum=MAX_IMAGE_SIZE,
-                step=64,
-                value=1024,
-            )
-        with gr.Row():
-            guidance_scale = gr.Slider(
-                label="Guidance Scale",
-                minimum=0.1,
-                maximum=6,
-                step=0.1,
-                value=3.0,
-            )
-            num_inference_steps = gr.Slider(
-                label="Number of inference steps",
-                minimum=1,
-                maximum=60,
-                step=1,
-                value=32,
-            )
-
-    gr.Examples(
-        examples=examples,
-        inputs=prompt,
-        cache_examples=False
-    )
-
-    use_negative_prompt.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_negative_prompt,
-        outputs=negative_prompt,
-        api_name=False,
-    )
-    
-    gr.on(
-        triggers=[
-            prompt.submit,
-            negative_prompt.submit,
-            run_button.click,
-        ],
-        fn=generate,
-        inputs=[
-            model_choice,
-            prompt,
-            negative_prompt,
-            use_negative_prompt,
-            style_selection,
-            seed,
-            width,
-            height,
-            guidance_scale,
-            num_inference_steps,
-            randomize_seed,
-            gr.State(value=True),  
-            num_images, 
-        ],
-        outputs=[result, seed],
-    )
-
-demo.queue(concurrency_count=3).launch(debug=True)

files/original-file.txt

@@ -1,300 +0,0 @@
-#!/usr/bin/env python
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-
-import os
-import random
-import uuid
-import gradio as gr
-import numpy as np
-from PIL import Image
-import spaces
-import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
-
-css = '''
-.gradio-container{max-width: 570px !important}
-h1{text-align:center}
-footer {
-    visibility: hidden
-}
-'''
-
-DESCRIPTIONXX = """
-    ## REALVISXL V5 + LIGHTNING ⚡
-"""
-examples = [
-
-    "Illustration of A starry night camp in the mountains, 4k, cinematic --ar 85:128 --v 6.0 --style raw",
-    "A delicious ceviche cheesecake slice, 4k, octane render, ray tracing, Ultra-High-Definition"
-]
-
-MODEL_OPTIONS = {
-    "REALVISXL V5.0": "SG161222/RealVisXL_V5.0",
-    #"LIGHTNING V5.0": "SG161222/RealVisXL_V5.0_Lightning",
-}
-
-MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
-USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
-ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
-BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
-
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
-def load_and_prepare_model(model_id):
-    pipe = StableDiffusionXLPipeline.from_pretrained(
-        model_id,
-        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-        use_safetensors=True,
-        add_watermarker=False,
-    ).to(device)
-    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    
-    if USE_TORCH_COMPILE:
-        pipe.compile()
-    
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    
-    return pipe
-
-# Preload and compile both models
-models = {key: load_and_prepare_model(value) for key, value in MODEL_OPTIONS.items()}
-
-MAX_SEED = np.iinfo(np.int32).max
-
-def save_image(img):
-    unique_name = str(uuid.uuid4()) + ".png"
-    img.save(unique_name)
-    return unique_name
-
-def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    return seed
-
-@spaces.GPU(duration=60, enable_queue=True)
-def generate(
-    model_choice: str,
-    prompt: str,
-    negative_prompt: str = "",
-    use_negative_prompt: bool = False,
-    seed: int = 1,
-    width: int = 1024,
-    height: int = 1024,
-    guidance_scale: float = 3,
-    num_inference_steps: int = 25,
-    randomize_seed: bool = False,
-    use_resolution_binning: bool = True, 
-    num_images: int = 1,  
-    progress=gr.Progress(track_tqdm=True),
-):
-    global models
-    pipe = models[model_choice]
-    
-    seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device=device).manual_seed(seed)
-
-    options = {
-        "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
-        "width": width,
-        "height": height,
-        "guidance_scale": guidance_scale,
-        "num_inference_steps": num_inference_steps,
-        "generator": generator,
-        "output_type": "pil",
-    }
-
-    if use_resolution_binning:
-        options["use_resolution_binning"] = True
-
-    images = []
-    for i in range(0, num_images, BATCH_SIZE):
-        batch_options = options.copy()
-        batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
-        if "negative_prompt" in batch_options:
-            batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
-        images.extend(pipe(**batch_options).images)
-
-    image_paths = [save_image(img) for img in images]
-    return image_paths, seed
-
-#def load_predefined_images():
- #   predefined_images = [
-   #     "assets/1.png",
-  #      "assets/2.png",
-    #    "assets/3.png",
-      #  "assets/4.png",
-     #   "assets/5.png",
-      #  "assets/6.png",
-       # "assets/7.png",
-        #"assets/8.png",
-        #"assets/9.png",
-    #]
-    #return predefined_images
-
-
-# def load_predefined_images():
-#     predefined_images = [
-#         "assets2/11.png",
-#         "assets2/22.png",
-#         "assets2/33.png",
-#         "assets2/44.png",
-#         "assets2/55.png",
-#         "assets2/66.png",
-#         "assets2/77.png",
-#         "assets2/88.png",
-#         "assets2/99.png",
-#     ]
-#     return predefined_images
-
-with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
-    gr.Markdown(DESCRIPTIONXX)
-    with gr.Row():
-        prompt = gr.Text(
-            label="Prompt",
-            show_label=False,
-            max_lines=1,
-            placeholder="Enter your prompt",
-            container=False,
-        )
-        run_button = gr.Button("Run", scale=0)
-    result = gr.Gallery(label="Result", columns=1, show_label=False) 
-
-    with gr.Row():
-        model_choice = gr.Dropdown(
-            label="Model Selection🔻",
-            choices=list(MODEL_OPTIONS.keys()),
-            value="REALVISXL V5.0"
-        )
-
-    with gr.Accordion("Advanced options", open=False, visible=True):
-        num_images = gr.Slider(
-            label="Number of Images",
-            minimum=1,
-            maximum=5,
-            step=1,
-            value=1,
-        )
-        with gr.Row():
-            with gr.Column(scale=1):
-                use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
-                negative_prompt = gr.Text(
-                    label="Negative prompt",
-                    max_lines=5,
-                    lines=4,
-                    placeholder="Enter a negative prompt",
-                    value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
-                    visible=True,
-                )
-        seed = gr.Slider(
-            label="Seed",
-            minimum=0,
-            maximum=MAX_SEED,
-            step=1,
-            value=0,
-        )
-        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-        with gr.Row():
-            width = gr.Slider(
-                label="Width",
-                minimum=512,
-                maximum=MAX_IMAGE_SIZE,
-                step=64,
-                value=1024,
-            )
-            height = gr.Slider(
-                label="Height",
-                minimum=512,
-                maximum=MAX_IMAGE_SIZE,
-                step=64,
-                value=1024,
-            )
-        with gr.Row():
-            guidance_scale = gr.Slider(
-                label="Guidance Scale",
-                minimum=0.1,
-                maximum=6,
-                step=0.1,
-                value=3.0,
-            )
-            num_inference_steps = gr.Slider(
-                label="Number of inference steps",
-                minimum=1,
-                maximum=60,
-                step=1,
-                value=32,
-            )
-
-    gr.Examples(
-        examples=examples,
-        inputs=prompt,
-        cache_examples=False
-    )
-
-    use_negative_prompt.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_negative_prompt,
-        outputs=negative_prompt,
-        api_name=False,
-    )
-    
-    gr.on(
-        triggers=[
-            prompt.submit,
-            negative_prompt.submit,
-            run_button.click,
-        ],
-        fn=generate,
-        inputs=[
-            model_choice,
-            prompt,
-            negative_prompt,
-            use_negative_prompt,
-            seed,
-            width,
-            height,
-            guidance_scale,
-            num_inference_steps,
-            randomize_seed,
-            num_images
-        ],
-        outputs=[result, seed],
-        api_name="run",
-    )
-
-
-    #gr.Markdown("### REALVISXL V5.0")
-    #predefined_gallery = gr.Gallery(label="REALVISXL V5.0", columns=3, show_label=False, value=load_predefined_images1())
-
-    #gr.Markdown("### LIGHTNING V5.0")
-    #predefined_gallery = gr.Gallery(label="LIGHTNING V5.0", columns=3, show_label=False, value=load_predefined_images())
-
-    gr.Markdown(
-    """
-    <div style="text-align: justify;">
-    ⚡Models used in the playground <a href="https://huggingface.co/SG161222/RealVisXL_V5.0">[REALVISXL V5.0]</a>, <a href="https://huggingface.co/SG161222/RealVisXL_V5.0_Lightning">[REALVISXL V5.0 LIGHTNING]</a> for image generation. Stable Diffusion XL piped (SDXL) model HF. This is the demo space for generating images using the Stable Diffusion XL models, with multiple different variants available.
-    </div>
-    """)
-
-    gr.Markdown(
-    """
-    <div style="text-align: justify;">
-    ⚡This is the demo space for generating images using Stable Diffusion XL with quality styles, different models, and types. Try the sample prompts to generate higher quality images. Try the sample prompts for generating higher quality images. 
-    <a href='https://huggingface.co/spaces/prithivMLmods/Top-Prompt-Collection' target='_blank'>Try prompts</a>.
-    </div>
-    """)
-
-    gr.Markdown(
-    """
-    <div style="text-align: justify;">
-    ⚠️ Users are accountable for the content they generate and are responsible for ensuring it meets appropriate ethical standards.
-    </div>
-    """) 
-        
-if __name__ == "__main__":
-    demo.queue(max_size=50).launch(show_api=False)

requirements.txt

@@ -1,10 +1,30 @@
-diffusers
+git+https://github.com/huggingface/diffusers.git
 torch
 torchvision
+xformers
 pipeline
 transformers
 accelerate
 safetensors
+sentencepiece
 spaces
+beautifulsoup4
+ftfy
 peft
-gradio
+protobuf
+invisible_watermark
+
+huggingface_hub
+pyyaml
+packaging
+numpy
+Pillow
+pytz
+python-dateutil
+click
+rich
+
+modin[all]
+optimum-intel
+openvino
+onnx