Update merged_app2.py

merged_app2.py

@@ -202,8 +202,8 @@ except ImportError:
     XFORMERS_AVAILABLE = False
     print("xformers not available - Using default attention")
 
-fill_pipe.enable_model_cpu_offload()
-fill_pipe.enable_vae_slicing()
+# fill_pipe.enable_model_cpu_offload()
+# fill_pipe.enable_vae_slicing()
 #fill_pipe.enable_xformers_memory_efficient_attention()
 
 
@@ -491,8 +491,8 @@ pipe = prepare_pipeline(
     dtype=dtype,
 )
 
-pipe.enable_model_cpu_offload()
-pipe.enable_vae_slicing()
+# pipe.enable_model_cpu_offload()
+# pipe.enable_vae_slicing()
 #pipe.enable_xformers_memory_efficient_attention()
 
 # Move models to device with consistent dtype
@@ -543,8 +543,8 @@ t2i_pipe = StableDiffusionPipeline(
     image_encoder=None
 )
 
-t2i_pipe.enable_model_cpu_offload()
-t2i_pipe.enable_vae_slicing()
+# t2i_pipe.enable_model_cpu_offload()
+# t2i_pipe.enable_vae_slicing()
 #t2i_pipe.enable_xformers_memory_efficient_attention()
 
 i2i_pipe = StableDiffusionImg2ImgPipeline(
@@ -559,8 +559,8 @@ i2i_pipe = StableDiffusionImg2ImgPipeline(
     image_encoder=None
 )
 
-i2i_pipe.enable_model_cpu_offload()
-i2i_pipe.enable_vae_slicing()
+# i2i_pipe.enable_model_cpu_offload()
+# i2i_pipe.enable_vae_slicing()
 #i2i_pipe.enable_xformers_memory_efficient_attention()
 
 @torch.inference_mode()
@@ -1742,7 +1742,10 @@ with gr.Blocks() as app:
 
                 with gr.Row():
                     with gr.Group():
-                        prompt = gr.Textbox(label="Prompt")
+                        with gr.Column():
+                            prompt = gr.Textbox(label="Prompt")
+                        with gr.Column():
+                            augment_prompt = gr.Button(value='+')
                         bg_source = gr.Radio(choices=[e.value for e in list(BGSource)[2:]],
                                             value=BGSource.LEFT.value,
                                             label="Lighting Preference (Initial Latent)", type='value')
@@ -1795,7 +1798,9 @@ with gr.Blocks() as app:
             #     outputs=[result_gallery, output_bg],
             #     run_on_click=True, examples_per_page=1024
             # )
+        augment_prompt.click(generate_description, inputs=[prompt,None], outputs=[prompt])
         ips = [extracted_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
+
         relight_button.click(fn=process_relight, inputs=ips, outputs=[result_gallery]).then(clear_memory, inputs=[], outputs=[])
         example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
         example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)