local

local_app.py

@@ -33,7 +33,7 @@ preprocessor = Preprocessor()
 preprocessor.load("NormalBae")
 
 if gr.NO_RELOAD:
-    # torch.cuda.max_memory_allocated(device="cuda")
+    torch.cuda.max_memory_allocated(device="cuda")
 
     # Controlnet Normal
     model_id = "lllyasviel/control_v11p_sd15_normalbae"
@@ -67,22 +67,22 @@ if gr.NO_RELOAD:
     vae = AutoencoderKL.from_single_file(vae_url, torch_dtype=torch.float16).to("cuda")
     vae.to(memory_format=torch.channels_last) 
     
-    pipe = StableDiffusionControlNetPipeline.from_pretrained(
-        base_model_id,
-        safety_checker=None,
+    # pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    #     base_model_id,
+    #     safety_checker=None,
+    #     controlnet=controlnet,
+    #     scheduler=scheduler,
+    #     vae=vae,
+    #     torch_dtype=torch.float16,
+    # ).to("cuda")
+    
+    pipe = StableDiffusionControlNetPipeline.from_single_file(
+        base_model_url,
         controlnet=controlnet,
         scheduler=scheduler,
         vae=vae,
         torch_dtype=torch.float16,
     ).to("cuda")
-    
-    # pipe = StableDiffusionControlNetPipeline.from_single_file(
-    #     base_model_url,
-    #     controlnet=controlnet,
-    #     scheduler=scheduler,
-    #     vae=vae,
-    #     torch_dtype=torch.float16,
-    # )
 
     pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="EasyNegativeV2.safetensors", token="EasyNegativeV2",)
     pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="badhandv4.pt", token="badhandv4")
@@ -105,7 +105,7 @@ if gr.NO_RELOAD:
     # @spaces.GPU(duration=12)
     # pipe.enable_xformers_memory_efficient_attention()
     # pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-    # pipe.unet.set_attn_processor(AttnProcessor2_0())
+    pipe.unet.set_attn_processor(AttnProcessor2_0())
     torch.cuda.empty_cache()
     gc.collect()
     print("Model Compiled!")
@@ -162,61 +162,65 @@ style_list = [
     },
     {
         "name": "Minimalistic",
-        "prompt": "Minimalistic"
+        "prompt": "Minimalist interior design, clean lines, neutral colors, uncluttered space, functional furniture, lots of natural light"
     },
     {
         "name": "Boho Chic",
-        "prompt": "boho chic"
+        "prompt": "Bohemian chic interior, eclectic mix of patterns and textures, vintage furniture, plants, woven textiles, warm earthy colors"
     },
     {
         "name": "Saudi Prince Gold",
-        "prompt": "saudi prince gold",
+        "prompt": "Opulent gold interior, luxurious ornate furniture, crystal chandeliers, rich fabrics, marble floors, intricate Arabic patterns"
     },
     {
         "name": "Modern Farmhouse",
-        "prompt": "modern farmhouse",
+        "prompt": "Modern farmhouse interior, rustic wood elements, shiplap walls, neutral color palette, industrial accents, cozy textiles"
     },
     {
         "name": "Neoclassical",
-        "prompt": "Neoclassical",
+        "prompt": "Neoclassical interior design, elegant columns, ornate moldings, symmetrical layout, refined furniture, muted color palette"
     },
     {
         "name": "Eclectic",
-        "prompt": "Eclectic",
+        "prompt": "Eclectic interior design, mix of styles and eras, bold color combinations, diverse furniture pieces, unique art objects"
     },
     {
         "name": "Parisian White",
-        "prompt": "Parisian White",
+        "prompt": "Parisian apartment interior, all-white color scheme, ornate moldings, herringbone wood floors, elegant furniture, large windows"
     },
     {
         "name": "Hollywood Glam",
-        "prompt": "Hollywood Glam",
+        "prompt": "Hollywood Regency interior, glamorous and luxurious, bold colors, mirrored surfaces, velvet upholstery, gold accents"
     },
     {
         "name": "Scandinavian",
-        "prompt": "Scandinavian",
+        "prompt": "Scandinavian interior design, light wood tones, white walls, minimalist furniture, cozy textiles, hygge atmosphere"
     },
     {
         "name": "Japanese",
-        "prompt": "Japanese",
+        "prompt": "Traditional Japanese interior, tatami mats, shoji screens, low furniture, zen garden view, minimalist decor, natural materials"
     },
     {
         "name": "Texas Cowboy",
-        "prompt": "Texas Cowboy",
+        "prompt": "Western cowboy interior, rustic wood beams, leather furniture, cowhide rugs, antler chandeliers, southwestern patterns"
     },
     {
         "name": "Midcentury Modern",
-        "prompt": "Midcentury Modern",
+        "prompt": "Mid-century modern interior, 1950s-60s style furniture, organic shapes, warm wood tones, bold accent colors, large windows"
     },
     {
         "name": "Beach",
-        "prompt": "Beach",
+        "prompt": "Coastal beach house interior, light blue and white color scheme, weathered wood, nautical accents, sheer curtains, ocean view"
     },
     {
-        "name": "The Matrix",
+        "name": "Retro Futurism",
         "prompt": "Neon (atompunk world) retro cyberpunk background",
     },
-]
+    {
+        "name": "The Matrix",
+        "prompt": "Futuristic cyberpunk interior, neon accent lighting, holographic plants, sleek black surfaces, advanced gaming setup, transparent screens, Blade Runner inspired decor, high-tech minimalist furniture, subtle Matrix green code effects, ceiling-to-floor digital displays, automated smart home features, hidden LED strips, metallic and glass materials, floating furniture pieces, ambient blue lighting"
+    }
+] 
 
 styles = {k["name"]: (k["prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
@@ -240,7 +244,7 @@ h3 {
     text-align: center;
     display:block;
 }
-.gradio-container{max-width: 1200px !important}
+.gradio-container{max-width: 1800px !important}
 footer {visibility: hidden}
 """
 with gr.Blocks(theme="bethecloud/storj_theme", css=css) as demo:
@@ -282,8 +286,8 @@ with gr.Blocks(theme="bethecloud/storj_theme", css=css) as demo:
             )
     #############################################################################
     # input text
-    with gr.Row():
-        gr.Text(label="Interior Design Style Examples", value="Eclectic, Maximalist, Bohemian, Scandinavian, Minimalist, Rustic, Modern Farmhouse, Contemporary, Luxury, Airbnb, Boho Chic, Midcentury Modern, Art Deco, Zen, Beach, Neoclassical, Industrial, Biophilic, Eco-friendly, Hollywood Glam, Parisian White, Saudi Prince Gold, French Country, Monster Energy Drink, Cyberpunk, Vaporwave, Baroque, etc.\n\nPro tip: add a color to customize it! You can also describe the furniture type.")
+    # with gr.Row():
+    #     gr.Text(label="Interior Design Style Examples", value="Eclectic, Maximalist, Bohemian, Scandinavian, Minimalist, Rustic, Modern Farmhouse, Contemporary, Luxury, Airbnb, Boho Chic, Midcentury Modern, Art Deco, Zen, Beach, Neoclassical, Industrial, Biophilic, Eco-friendly, Hollywood Glam, Parisian White, Saudi Prince Gold, French Country, Monster Energy Drink, Cyberpunk, Vaporwave, Baroque, etc.\n\nPro tip: add a color to customize it! You can also describe the furniture type.")
     with gr.Column():
         prompt = gr.Textbox(
             label="Custom Prompt",