init

app.py

@@ -31,17 +31,28 @@ if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+# pipe.load_lora_weights(hf_hub_download("RED-AIGC/TDD", "TDD-FLUX.1-dev-lora-beta.safetensors"))
 pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to(device="cuda", dtype=torch.bfloat16)
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+css = """
+h1 {
+    text-align: center;
+    display:block;
+}
+.gradio-container {
+  max-width: 70.5rem !important;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
     gr.Markdown(
         """
-        <div style="text-align: center; max-width: 650px; margin: 0 auto;">
-            <h1 style="font-size: 2.5rem; font-weight: 700; margin-bottom: 1rem; display: contents;">Hyper-FLUX-8steps-LoRA</h1>
-            <p style="font-size: 1rem; margin-bottom: 1.5rem;">AutoML team from ByteDance</p>
-        </div>
+        # FLUX.1-dev(beta) distilled by ✨Target-Driven Distillation✨
+        Compared to Hyper-FLUX, the beta version of TDD has its parameters reduced by half(600M), resulting in more realistic details. Due to the limitations of the machine configuration, there are still many imperfections in the test version. The official version is still being optimized and is expected to be released after the Mid-Autumn Festival. TDD is also available for distilling video generation models. This space presents TDD-distilled [FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev).
+        [**Project Page**](https://redaigc.github.io/TDD/) **|** [**Paper**](https://arxiv.org/abs/2409.01347) **|** [**Code**](https://github.com/RedAIGC/Target-Driven-Distillation) **|** [**Model**](https://huggingface.co/RED-AIGC/TDD) **|** [🤗 **TDD-SDXL Demo**](https://huggingface.co/spaces/RED-AIGC/TDD) **|** [🤗 **TDD-SVD Demo**](https://huggingface.co/spaces/RED-AIGC/SVD-TDD)
+        The codes of this space are built on [Hyper-SD](https://huggingface.co/spaces/ByteDance/Hyper-FLUX-8Steps-LoRA) and we acknowledge their contribution.
         """
     )
 
@@ -49,8 +60,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=3):
             with gr.Group():
                 prompt = gr.Textbox(
-                    label="Your Image Description",
-                    placeholder="E.g., A serene landscape with mountains and a lake at sunset",
+                    label="Prompt",
+                    placeholder="E.g., portrait photo of a girl, photograph, highly detailed face, depth of field, moody light, golden hour, style by Dan Winters, Russell James, Steve McCurry, centered, extremely detailed, Nikon D850, award winning photography",
                     lines=3
                 )
                 
@@ -61,10 +72,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                             width = gr.Slider(label="Width", minimum=256, maximum=1152, step=64, value=1024)
                         
                         with gr.Row():
-                            steps = gr.Slider(label="Inference Steps", minimum=6, maximum=25, step=1, value=8)
-                            scales = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=5.0, step=0.1, value=3.5)
+                            steps = gr.Slider(label="Inference Steps", minimum=4, maximum=10, step=1, value=8)
+                            scales = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=3.5, step=0.1, value=2.0)
                         
-                        seed = gr.Number(label="Seed (for reproducibility)", value=3413, precision=0)
+                        seed = gr.Number(label="Seed", value=3413, precision=0)
                 
                 generate_btn = gr.Button("Generate Image", variant="primary", scale=1)