hysts commited on
Commit
534f2b8
1 Parent(s): 700bc3e

Hide the "remove GPU after training" button.

Browse files

Training of a LoRA DreamBooth model often only takes about 10 minutes
and this feature is not very useful for this Space.

Files changed (1) hide show
  1. 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():