size

app.py

@@ -406,7 +406,7 @@ def get_gallery_selection(evt: gr.SelectData):
 
 # --- Generation Functions ---
 @spaces.GPU(duration=120)
-def generate_t2i(prompt, num_inference_steps, num_images, current_color, height, width):
+def generate_t2i(prompt, num_inference_steps, num_images, current_color, height=512, width=512):
     """
     Generates multiple images for Text-to-Image and returns a gallery.
     """
@@ -659,9 +659,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
                     t2i_steps = gr.Slider(10, 200, 50, step=1, label="Inference Steps")
                     # Added: Number of Images Slider
                     t2i_num_images = gr.Slider(1, 9, 3, step=1, label="Number of Images")
-                    with gr.Row(): 
-                        t2i_height = gr.Slider(256, 4096, value=512, step=64, label="Height")
-                        t2i_width = gr.Slider(256, 4096, value=512, step=64, label="Width")
+                    # with gr.Row(): 
+                    #     t2i_height = gr.Slider(512, 1024, value=512, step=64, label="Height")
+                    #     t2i_width = gr.Slider(512, 1024, value=512, step=64, label="Width")
                     t2i_btn = gr.Button("Generate", variant="primary")
                 with gr.Column(scale=2):
                     # Changed: Image to Gallery
@@ -675,11 +675,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
                             t2i_dl = gr.DownloadButton(label="Download All (.zip)")
             t2i_gal = gr.Gallery(value=t2i_examples, label="Examples", columns=6, height="auto")
 
-            t2i_btn.click(
-                generate_t2i, 
-                inputs=[t2i_prompt, t2i_steps, t2i_num_images, t2i_color, t2i_height, t2i_width], 
-                outputs=[t2i_gallery_out, t2i_dl, t2i_raw_state, t2i_colorbar]
-            )
+            t2i_btn.click(generate_t2i, [t2i_prompt, t2i_steps, t2i_num_images, t2i_color], [t2i_gallery_out, t2i_dl, t2i_raw_state, t2i_colorbar])
+            # t2i_btn.click(
+            #     generate_t2i, 
+            #     inputs=[t2i_prompt, t2i_steps, t2i_num_images, t2i_color, t2i_height, t2i_width], 
+            #     outputs=[t2i_gallery_out, t2i_dl, t2i_raw_state, t2i_colorbar]
+            # )
             # Reuse update_gallery_color since state is now a list
             t2i_color.change(update_gallery_color, [t2i_raw_state, t2i_color], [t2i_gallery_out, t2i_colorbar])
             t2i_gal.select(fn=get_gallery_selection, inputs=None, outputs=t2i_prompt)

models/pipeline_ddpm_text_encoder.py

@@ -164,4 +164,5 @@ class DDPMPipeline(DiffusionPipeline):
         if not return_dict:
             return (image,)
 
-        return ImagePipelineOutput(images=image)
+        return ImagePipelineOutput(images=image)
+