Add training/inference time info

app_inference.py

@@ -121,6 +121,8 @@ def create_inference_demo(pipe: InferencePipeline,
 
                 gr.Markdown('''
                 - After training, you can press "Reload Model List" button to load your trained model names.
+                - It takes a few minutes to download model first.
+                - Expected time to generate an 8-frame video: 24 seconds with A10G, 70 seconds with T4, (10 seconds with A100)
                 ''')
             with gr.Column():
                 result = gr.Video(label='Result')

app_training.py

@@ -87,8 +87,7 @@ def create_training_demo(trainer: Trainer,
                 gr.Markdown('''
                     - The base model must be a model that is compatible with [diffusers](https://github.com/huggingface/diffusers) library.
                     - It takes a few minutes to download the base model first.
-                    - It will take about 8 minutes to train for 300 steps with an A10G GPU.
-                    - It will take about 4 minutes with an A100 GPU.
+                    - Expected time to train a model for 300 steps: 8 minutes with A10G, 20 minutes with T4, (4 minutes with A100)
                     - It takes a few minutes to upload your trained model.
                     - You may want to try a small number of steps first, like 1, to see if everything works fine in your environment.
                     - You can check the training status by pressing the "Open logs" button if you are running this on your Space.