add debug print + add size silder

animatediff/models/motion_module.py

@@ -472,6 +472,7 @@ class CrossAttention(nn.Module):
         return hidden_states
 
     def _memory_efficient_attention_pt20(self, query, key, value, attention_mask):
+        print('Use PT20 Attention')
         query = query.contiguous()
         key = key.contiguous()
         value = value.contiguous()

app-huggingface.py

@@ -346,6 +346,7 @@ class AnimateController:
         style,
         with_text=False,
         text_idx=0,
+        max_size=512,
         progress=gr.Progress(),
 
     ):
@@ -374,7 +375,7 @@ class AnimateController:
         print(f'Seed: {seed}')
 
         pipeline = self.pipeline_dict[style]
-        init_img, h, w = preprocess_img(init_img)
+        init_img, h, w = preprocess_img(init_img, max_size)
         print(f'img size: {h, w}')
 
         sample = pipeline(
@@ -517,6 +518,9 @@ def ui():
                 motion_scale_silder = gr.Slider(
                     label='Motion Scale (Larger value means larger motion but less identity consistency)',
                     value=1, step=1, minimum=1, maximum=len(RANGE_LIST))
+                max_size_silder = gr.Slider(
+                    label='Max size (The long edge of the input image will be resized to this value, larger value means slower inference speed)',
+                    value=512, step=64, minimum=512, maximum=1024)
 
                 with gr.Accordion('Advance Options', open=False):
                     negative_prompt_textbox = gr.Textbox(
@@ -567,6 +571,7 @@ def ui():
                 seed_textbox,
                 ip_adapter_scale,
                 style_dropdown,
+                max_size_silder,
             ],
             outputs=[result_video, download])