Spaces:
Build error
Build error
Nupur Kumari
commited on
Commit
•
7060e57
1
Parent(s):
98f17d6
custom-diffusion-space
Browse files
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
|
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 |
|