print xla env vars

app.py

@@ -9,6 +9,8 @@ import gradio as gr
 
 from makeavid_sd.inference import InferenceUNetPseudo3D, FlaxDPMSolverMultistepScheduler, jnp
 
+print(os.environ.get('XLA_PYTHON_CLIENT_PREALLOCATE', 'NotSet'))
+print(os.environ.get('XLA_PYTHON_CLIENT_ALLOCATOR', 'NotSet'))
 
 _preheat: bool = False
 
@@ -226,7 +228,7 @@ with gr.Blocks(title = 'Make-A-Video Stable Diffusion JAX', analytics_enabled =
                     value = 12
             )
         with gr.Column(variant = variant):
-            no_gpu = gr.Markdown('**Until a GPU is assigned expect extremely long runtimes up to 1h+**')
+            #no_gpu = gr.Markdown('**Until a GPU is assigned expect extremely long runtimes up to 1h+**')
             will_trigger = gr.Markdown('')
             patience = gr.Markdown('')
             image_output = gr.Image(