fix sd

app.py

@@ -455,6 +455,8 @@ def run_fn(
     num_images = len(images)
     if num_images > 100:
         return super_duper_long_run(model, images, **kwargs)
+    if 'diffusion' in model_name.lower():
+        return super_duper_long_run(model, images, **kwargs)
     if recursion:
         return longer_run(model, images, **kwargs)
     if num_images > 50:
@@ -755,6 +757,16 @@ with demo:
                 
         gr.Markdown('---')
 
+        with gr.Row():
+            with gr.Column(scale=5, min_width=200):
+                gr.Markdown('### Output (Recursion #1)')
+                l1_gallery = gr.Gallery(value=[], label="Recursion #1", show_label=False, elem_id="ncut_l1", columns=[3], rows=[5], object_fit="contain", height="auto")
+            with gr.Column(scale=5, min_width=200):
+                gr.Markdown('### Output (Recursion #2)')
+                l2_gallery = gr.Gallery(value=[], label="Recursion #2", show_label=False, elem_id="ncut_l2", columns=[3], rows=[5], object_fit="contain", height="auto")
+            with gr.Column(scale=5, min_width=200):
+                gr.Markdown('### Output (Recursion #3)')
+                l3_gallery = gr.Gallery(value=[], label="Recursion #3", show_label=False, elem_id="ncut_l3", columns=[3], rows=[5], object_fit="contain", height="auto")
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
                 input_gallery, submit_button, clear_images_button = make_input_images_section()
@@ -781,16 +793,6 @@ with demo:
                 num_eig_slider.visible = False
                 affinity_focal_gamma_slider.visible = False
                 # logging text box
-        with gr.Row():
-            with gr.Column(scale=5, min_width=200):
-                gr.Markdown('### Output (Recursion #1)')
-                l1_gallery = gr.Gallery(value=[], label="Recursion #1", show_label=False, elem_id="ncut_l1", columns=[3], rows=[5], object_fit="contain", height="auto")
-            with gr.Column(scale=5, min_width=200):
-                gr.Markdown('### Output (Recursion #2)')
-                l2_gallery = gr.Gallery(value=[], label="Recursion #2", show_label=False, elem_id="ncut_l2", columns=[3], rows=[5], object_fit="contain", height="auto")
-            with gr.Column(scale=5, min_width=200):
-                gr.Markdown('### Output (Recursion #3)')
-                l3_gallery = gr.Gallery(value=[], label="Recursion #3", show_label=False, elem_id="ncut_l3", columns=[3], rows=[5], object_fit="contain", height="auto")
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
                 gr.Markdown(' ')

requirements.txt

@@ -6,6 +6,7 @@ decord
 transformers
 datasets
 diffusers
+accelerate
 hydra-core==1.3.2
 iopath==0.1.10
 pillow==9.4.0