remove trigger check, as it messes up the time estimate

app.py

@@ -89,6 +89,8 @@ def generate(
 def check_if_compiled(image, inference_steps, height, width, num_frames, message):
     height = int(height)
     width = int(width)
+    height = (height // 64) * 64
+    width = (width // 64) * 64
     hint_image = image
     if (hint_image is None, inference_steps, height, width, num_frames) in _seen_compilations:
         return ''
@@ -100,7 +102,7 @@ if _preheat:
     generate(
             prompt = 'preheating the oven',
             neg_prompt = '',
-            image = { 'image': None, 'mask': None },
+            image = None,
             inference_steps = 20,
             cfg = 12.0,
             seed = 0
@@ -109,7 +111,7 @@ if _preheat:
     dada = generate(
             prompt = 'Entertaining the guests with sailor songs played on an old harmonium.',
             neg_prompt = '',
-            image = { 'image': Image.new('RGB', size = (512, 512), color = (0, 0, 0)), 'mask': None },
+            image = Image.new('RGB', size = (512, 512), color = (0, 0, 0)),
             inference_steps = 20,
             cfg = 12.0,
             seed = 0
@@ -227,29 +229,22 @@ with gr.Blocks(title = 'Make-A-Video Stable Diffusion JAX', analytics_enabled =
             )
         with gr.Column(variant = variant):
             #no_gpu = gr.Markdown('**Until a GPU is assigned expect extremely long runtimes up to 1h+**')
-            will_trigger = gr.Markdown('')
-            patience = gr.Markdown('')
+            #will_trigger = gr.Markdown('')
+            patience = gr.Markdown('**Please be patient. The model might have to compile with current parameters.**')
             image_output = gr.Image(
                     label = 'Output',
                     value = 'example.webp',
                     interactive = False
             )
-    trigger_inputs =  [ image_input, inference_steps_input, height_input, width_input, num_frames_input ]
-    trigger_check_fun = partial(check_if_compiled, message = 'Current parameters will trigger compilation.')
-    height_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
-    width_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
-    num_frames_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
-    image_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
-    inference_steps_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
-    will_trigger.value = trigger_check_fun(image_input.value, inference_steps_input.value, height_input.value, width_input.value, num_frames_input.value)
+    #trigger_inputs = [ image_input, inference_steps_input, height_input, width_input, num_frames_input ]
+    #trigger_check_fun = partial(check_if_compiled, message = 'Current parameters will trigger compilation.')
+    #height_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
+    #width_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
+    #num_frames_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
+    #image_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
+    #inference_steps_input.change(fn = trigger_check_fun, inputs = trigger_inputs, outputs = will_trigger)
+    #will_trigger.value = trigger_check_fun(image_input.value, inference_steps_input.value, height_input.value, width_input.value, num_frames_input.value)
     ev = submit_button.click(
-        fn = partial(
-                check_if_compiled,
-                message = 'Please be patient. The model has to be compiled with current parameters.'
-        ),
-        inputs = trigger_inputs,
-        outputs = patience
-    ).then(
         fn = generate,
         inputs = [
                 prompt_input,
@@ -265,10 +260,6 @@ with gr.Blocks(title = 'Make-A-Video Stable Diffusion JAX', analytics_enabled =
         ],
         outputs = image_output,
         postprocess = False
-    ).then(
-        fn = trigger_check_fun,
-        inputs = trigger_inputs,
-        outputs = will_trigger
     )
     #cancel_button.click(fn = lambda: None, cancels = ev)
 

requirements.txt

@@ -6,5 +6,5 @@ einops
 -f https://download.pytorch.org/whl/cpu/torch
 torch[cpu]
 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-jax[cuda11_cudnn82] #jax[cuda11_cudnn86] #jax[cuda11_cudnn805]
+jax[cuda11_pip] #jax[cuda11_cudnn82] #jax[cuda11_cudnn86] #jax[cuda11_cudnn805]
 flax