hysts
commited on
Commit
·
534f2b8
1
Parent(s):
700bc3e
Hide the "remove GPU after training" button.
Browse filesTraining of a LoRA DreamBooth model often only takes about 10 minutes
and this feature is not very useful for this Space.
- app_training.py +2 -1
app_training.py
CHANGED
|
@@ -95,7 +95,8 @@ def create_training_demo(trainer: Trainer,
|
|
| 95 |
remove_gpu_after_training = gr.Checkbox(
|
| 96 |
label='Remove GPU after training',
|
| 97 |
value=False,
|
| 98 |
-
interactive=bool(os.getenv('SPACE_ID'))
|
|
|
|
| 99 |
run_button = gr.Button('Start Training')
|
| 100 |
|
| 101 |
with gr.Box():
|
|
|
|
| 95 |
remove_gpu_after_training = gr.Checkbox(
|
| 96 |
label='Remove GPU after training',
|
| 97 |
value=False,
|
| 98 |
+
interactive=bool(os.getenv('SPACE_ID')),
|
| 99 |
+
visible=False)
|
| 100 |
run_button = gr.Button('Start Training')
|
| 101 |
|
| 102 |
with gr.Box():
|