Nupur Kumari commited on
Commit
7060e57
1 Parent(s): 98f17d6

custom-diffusion-space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ def create_training_demo(trainer: Trainer,
101
  gr.Markdown('''
102
  - Only enable one of "Train Text Encoder" or "modifier token" or None.
103
  - It will take about ~10 minutes to train for 1000 steps and ~21GB on a 3090 GPU.
104
- - Enable gradient checkpointing to save memory (~14GB) at the expense of slower backward pass.
105
  - Note that your trained models will be deleted when the second training is started. You can upload your trained model in the "Upload" tab.
106
  ''')
107
 
 
101
  gr.Markdown('''
102
  - Only enable one of "Train Text Encoder" or "modifier token" or None.
103
  - It will take about ~10 minutes to train for 1000 steps and ~21GB on a 3090 GPU.
104
+ - Enable gradient checkpointing for lower memory requirements (~14GB) at the expense of slower backward pass.
105
  - Note that your trained models will be deleted when the second training is started. You can upload your trained model in the "Upload" tab.
106
  ''')
107