add: sdxl app + dependencies

app.py

@@ -1,7 +1,121 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+import torch
+import numpy as np
+import modin.pandas as pd
+from PIL import Image
+from diffusers import DiffusionPipeline
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+import wandb
+from wandb_addons.diffusers import get_wandb_callback
+
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = None
+
+if torch.cuda.is_available():
+    torch.cuda.empty_cache()
+
+    pipeline = DiffusionPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
+    )
+    pipeline = pipeline.to("device")
+    pipeline.enable_xformers_memory_efficient_attention()
+
+else:
+    pipeline = DiffusionPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-base-1.0", use_safetensors=True
+    )
+    pipeline = pipeline.to(device)
+    pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+
+
+def generate_image(
+    wandb_project,
+    wandb_entity,
+    wandb_api_key,
+    prompt,
+    negative_prompt,
+    height,
+    width,
+    num_inference_steps,
+    guidance_scale,
+    seed,
+    guidance_rescale,
+):
+    wandb.login(key=wandb_api_key)
+    generator = torch.Generator(device="cuda").manual_seed(seed)
+    configs = {
+        "guidance_rescale": 0.0,
+        "seed": seed,
+    }
+    callback = get_wandb_callback(
+        pipeline,
+        prompt=prompt,
+        negative_prompt=negative_prompt,
+        wandb_project=wandb_project,
+        wandb_entity=wandb_entity,
+        configs=configs,
+    )
+    image = pipeline(
+        prompt,
+        negative_prompt=negative_prompt,
+        generator=generator,
+        height=height,
+        width=width,
+        callback=callback,
+        eta=configs["eta"],
+        guidance_rescale=configs["guidance_rescale"],
+    ).images[0]
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+    return image
+
+
+gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(label="Wandb Project Name", value="Stable-Diffusion-XL"),
+        gr.Textbox(label="Wandb Entity Name (Username/Team-name)"),
+        gr.Textbox(label="Wandb API Key", type="password"),
+        gr.Textbox(
+            label="Prompt",
+            lines=3,
+            value="a woman in orange, extremely detailed digital painting, in the style of Fenghua Zhong and Ruan Jia and jeremy lipking and Peter Mohrbacher, mystical colors, rim light, beautiful Lighting, 8k, stunning scene, raytracing, octane, trending on artstation",
+        ),
+        gr.Textbox(
+            lablel="Negative Prompt",
+            lines=3,
+            value="blurry, plastic, grainy, duplicate, deformed , disfigured, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limb, disconnected limb, mutated hands and fingers, text, name, signature, watermark, worst quality, jpeg artifacts, boring composition, uninteresting",
+        ),
+        gr.Slider(minimum=512, maximum=1024, value=1024, step=128, label="Height"),
+        gr.Slider(minimum=512, maximum=1024, value=1024, step=128, label="Width"),
+        gr.Slider(
+            minimum=10,
+            maximum=100,
+            value=50,
+            step=128,
+            label="Number of Inference Steps",
+        ),
+        gr.Slider(
+            minimum=1.0,
+            maximum=15.0,
+            value=5.0,
+            step=0.25,
+            label="Guidance Scale: How Closely the model follows the Prompt",
+        ),
+        gr.Slider(
+            minimum=0, step=1, maximum=999999999999999999, randomize=True, label="Seed"
+        ),
+        gr.Slider(
+            minimum=0.0,
+            maximum=1.0,
+            value=0.7,
+            step=0.1,
+            label="Guidance Rescale Factor",
+        ),
+    ],
+    outputs="image",
+    title="Reproducible Stable Diffusion XL",
+    description="Generate images using Stable Diffusion.",
+)

requirements.txt

@@ -0,0 +1,10 @@
+transformers
+diffusers
+accelerate
+torch
+ftfy
+modin[all]
+xformers
+invisible_watermark
+weave
+wandb-addons @ git+https://github.com/soumik12345/wandb-addons.git@292f049937424a1a40b833c194e0edeb9bed446d