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():
|