multi_decoder

app.py

@@ -14,6 +14,10 @@ LOW_MEMORY = os.getenv("LOW_MEMORY", "0") == "1"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16
 
+multi_decoder = (
+    torch.cuda.get_device_properties(0).total_memory < 18 * 1024 * 1024 * 1024
+)
+
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=dtype)
 pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
@@ -87,7 +91,7 @@ def predict(
         cosine_scale_2=1,
         cosine_scale_3=1,
         sigma=0.8,
-        multi_decoder=True,
+        multi_decoder=multi_decoder,
         show_image=False,
         lowvram=LOW_MEMORY,
     )
@@ -112,13 +116,17 @@ css = """
 with gr.Blocks(css=css) as demo:
     gr.Markdown(
         """
-# Zoom and Enhance with DemoFusion SDXL
+# Enhance This  
+### DemoFusion SDXL
 
-[DemoFusion](https://github.com/PRIS-CV/DemoFusion) enables higher-resolution image generation.  
+[DemoFusion](https://ruoyidu.github.io/demofusion/demofusion.html) enables higher-resolution image generation.  
 You can upload an initial image and prompt to generate an enhanced version. 
 [Duplicate Space](https://huggingface.co/spaces/radames/Enhance-This-DemoFusion-SDXL?duplicate=true) to avoid the queue.  
+GPU Time Comparison: T4: - A10G: ~175s A100: RTX 4090: ~88.8s
+
 <small>
-*Note*: The author advises against the term "super resolution" because it's more like image-to-image generation than enhancement, but it's still a lot of fun!
+<b>Notes</b> The author advises against the term "super resolution" because it's more like image-to-image generation than enhancement, but it's still a lot of fun!
+
 </small>
         """,
         elem_id="intro",
@@ -172,8 +180,8 @@ You can upload an initial image and prompt to generate an enhanced version.
                 "./examples/cybetruck.jpeg",
                 "photo of tesla cybertruck futuristic car 8k high definition on a sand dune in mars, future",
                 "blurry, ugly, duplicate, poorly drawn, deformed, mosaic",
-                9257959681232992980,
-                3,
+                23232234234,
+                2,
             ],
             [
                 "./examples/jesus.png",
@@ -186,8 +194,8 @@ You can upload an initial image and prompt to generate an enhanced version.
                 "./examples/anna-sullivan-DioLM8ViiO8-unsplash.jpg",
                 "A crowded stadium with enthusiastic fans watching a daytime sporting event, the stands filled with colorful attire and the sun casting a warm glow",
                 "blurry, ugly, duplicate, poorly drawn, deformed, mosaic",
-                8398712905087378000,
-                3,
+                12412356145,
+                2,
             ],
         ],
         inputs=inputs,

requirements.txt

@@ -10,4 +10,4 @@ accelerate
 invisible-watermark
 huggingface-hub
 hf-transfer
-https://huggingface.co/datasets/radames/gradio-components/resolve/main/gradio_imageslider-0.0.13-py3-none-any.whl
+gradio_imageslider==0.0.14