hysts HF staff commited on
Commit
6558e17
1 Parent(s): acc152e

Add training/inference time info

Browse files
Files changed (2) hide show
  1. app_inference.py +2 -0
  2. app_training.py +1 -2
app_inference.py CHANGED
@@ -121,6 +121,8 @@ def create_inference_demo(pipe: InferencePipeline,
121
 
122
  gr.Markdown('''
123
  - After training, you can press "Reload Model List" button to load your trained model names.
 
 
124
  ''')
125
  with gr.Column():
126
  result = gr.Video(label='Result')
 
121
 
122
  gr.Markdown('''
123
  - After training, you can press "Reload Model List" button to load your trained model names.
124
+ - It takes a few minutes to download model first.
125
+ - Expected time to generate an 8-frame video: 24 seconds with A10G, 70 seconds with T4, (10 seconds with A100)
126
  ''')
127
  with gr.Column():
128
  result = gr.Video(label='Result')
app_training.py CHANGED
@@ -87,8 +87,7 @@ def create_training_demo(trainer: Trainer,
87
  gr.Markdown('''
88
  - The base model must be a model that is compatible with [diffusers](https://github.com/huggingface/diffusers) library.
89
  - It takes a few minutes to download the base model first.
90
- - It will take about 8 minutes to train for 300 steps with an A10G GPU.
91
- - It will take about 4 minutes with an A100 GPU.
92
  - It takes a few minutes to upload your trained model.
93
  - You may want to try a small number of steps first, like 1, to see if everything works fine in your environment.
94
  - You can check the training status by pressing the "Open logs" button if you are running this on your Space.
 
87
  gr.Markdown('''
88
  - The base model must be a model that is compatible with [diffusers](https://github.com/huggingface/diffusers) library.
89
  - It takes a few minutes to download the base model first.
90
+ - Expected time to train a model for 300 steps: 8 minutes with A10G, 20 minutes with T4, (4 minutes with A100)
 
91
  - It takes a few minutes to upload your trained model.
92
  - You may want to try a small number of steps first, like 1, to see if everything works fine in your environment.
93
  - You can check the training status by pressing the "Open logs" button if you are running this on your Space.